llm-dialog-manager 0.4.5__py3-none-any.whl → 0.4.6__py3-none-any.whl

llm_dialog_manager/__init__.py

@@ -1,4 +1,4 @@
 from .chat_history import ChatHistory
 from .agent import Agent
 
-__version__ = "0.4.5" 
+__version__ = "0.4.6" 

llm_dialog_manager/agent.py

@@ -394,16 +394,21 @@ def completion(model: str, messages: List[Dict[str, Union[str, List[Union[str, I
             if model.endswith("-openai"):
                 model = model[:-7]  # Remove last 7 characters ("-openai")
             client = openai.OpenAI(api_key=api_key, base_url=base_url)
-            # Set response_format based on json_format
-            response_format = {"type": "json_object"} if json_format else {"type": "plain_text"}
-
-            response = client.chat.completions.create(
-                model=model,
-                messages=formatted_messages,
-                max_tokens=max_tokens,
-                temperature=temperature,
-                response_format=response_format  # Added response_format
-            )
+            
+            # Create base parameters
+            params = {
+                "model": model,
+                "messages": formatted_messages,
+            }
+            
+            # Add optional parameters
+            if json_format:
+                params["response_format"] = {"type": "json_object"}
+            if not ("o1" in model or "o3" in model):
+                params["max_tokens"] = max_tokens
+                params["temperature"] = temperature
+
+            response = client.chat.completions.create(**params)
             return response.choices[0].message.content
 
         # Release the API key after successful use

{llm_dialog_manager-0.4.5.dist-info → llm_dialog_manager-0.4.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: llm_dialog_manager
-Version: 0.4.5
+Version: 0.4.6
 Summary: A Python package for managing LLM chat conversation history
 Author-email: xihajun <work@2333.fun>
 License: MIT

llm_dialog_manager-0.4.6.dist-info/RECORD

@@ -0,0 +1,9 @@
+llm_dialog_manager/__init__.py,sha256=klLFvHayR7ew1Oh9xyhAXdXnfs82YnFUEFzw0YvxKJI,86
+llm_dialog_manager/agent.py,sha256=NVQKIMebl4cYkqMaBceZ3qs1vYhq1bum9okAn8VcfCg,27680
+llm_dialog_manager/chat_history.py,sha256=DKKRnj_M6h-4JncnH6KekMTghX7vMgdN3J9uOwXKzMU,10347
+llm_dialog_manager/key_manager.py,sha256=shvxmn4zUtQx_p-x1EFyOmnk-WlhigbpKtxTKve-zXk,4421
+llm_dialog_manager-0.4.6.dist-info/LICENSE,sha256=vWGbYgGuWpWrXL8-xi6pNcX5UzD6pWoIAZmcetyfbus,1064
+llm_dialog_manager-0.4.6.dist-info/METADATA,sha256=-qTRYkfAJMJCQTkRqNrtHjUuN-xGLhLR4CJvSJURgeg,4194
+llm_dialog_manager-0.4.6.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+llm_dialog_manager-0.4.6.dist-info/top_level.txt,sha256=u2EQEXW0NGAt0AAHT7jx1odXZ4rZfjcgbmJhvKFuMkI,19
+llm_dialog_manager-0.4.6.dist-info/RECORD,,

llm_dialog_manager-0.4.5.dist-info/RECORD

@@ -1,9 +0,0 @@
-llm_dialog_manager/__init__.py,sha256=Vai6V5cLO8X7dkHgnALzKyhtDYdPhkd9oD0HPjzdqZ8,86
-llm_dialog_manager/agent.py,sha256=cYK3SMa1zPeSP4QoSxfceQ3YLQYqBTNwDE_ZJdBi7pM,27595
-llm_dialog_manager/chat_history.py,sha256=DKKRnj_M6h-4JncnH6KekMTghX7vMgdN3J9uOwXKzMU,10347
-llm_dialog_manager/key_manager.py,sha256=shvxmn4zUtQx_p-x1EFyOmnk-WlhigbpKtxTKve-zXk,4421
-llm_dialog_manager-0.4.5.dist-info/LICENSE,sha256=vWGbYgGuWpWrXL8-xi6pNcX5UzD6pWoIAZmcetyfbus,1064
-llm_dialog_manager-0.4.5.dist-info/METADATA,sha256=TbPdPjwrncfdYIJjFTzOoaWp5p90zYzpzsazVSK3F_8,4194
-llm_dialog_manager-0.4.5.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-llm_dialog_manager-0.4.5.dist-info/top_level.txt,sha256=u2EQEXW0NGAt0AAHT7jx1odXZ4rZfjcgbmJhvKFuMkI,19
-llm_dialog_manager-0.4.5.dist-info/RECORD,,