habib926653 commited on
Commit
509509a
·
verified ·
1 Parent(s): acede7f

Update structured_output_extractor.py

Browse files
Files changed (1) hide show
  1. structured_output_extractor.py +2 -1
structured_output_extractor.py CHANGED
@@ -23,7 +23,8 @@ class StructuredOutputExtractor:
23
  self.response_schema = response_schema
24
 
25
  # Initialize language model (provider and API keys come from constants.py)
26
- self.llm = ChatGroq(model="llama-3.3-70b-versatile")
 
27
 
28
  # Bind the model with structured output capability
29
  self.structured_llm = self.llm.with_structured_output(response_schema)
 
23
  self.response_schema = response_schema
24
 
25
  # Initialize language model (provider and API keys come from constants.py)
26
+ # self.llm = ChatGroq(model="llama-3.3-70b-versatile") # token limit 100k tokens
27
+ self.llm = ChatGroq(model="deepseek-r1-distill-llama-70b") # currently no limit per day
28
 
29
  # Bind the model with structured output capability
30
  self.structured_llm = self.llm.with_structured_output(response_schema)