Update structured_output_extractor.py
Browse files
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)
|