diff --git a/utils/llm_client.py b/utils/llm_client.py index ea9cf59..760c5eb 100644 --- a/utils/llm_client.py +++ b/utils/llm_client.py @@ -123,6 +123,7 @@ class LLMClient: tools=[ANALYSIS_TOOL], tool_choice={"type": "function", "function": {"name": "report_analysis"}}, temperature=0.1, + max_tokens=1024, ) choice = response.choices[0] @@ -255,6 +256,7 @@ class LLMClient: tools=[ANALYSIS_TOOL], tool_choice={"type": "function", "function": {"name": "report_analysis"}}, temperature=0.1, + max_tokens=1024, ) choice = response.choices[0]