llm-dialog-manager 0.4.5__tar.gz → 0.4.6__tar.gz

{llm_dialog_manager-0.4.5 → llm_dialog_manager-0.4.6}/PKG-INFO

@@ -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.5 → llm_dialog_manager-0.4.6}/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-0.4.5 → llm_dialog_manager-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 → llm_dialog_manager-0.4.6}/llm_dialog_manager.egg-info/PKG-INFO

@@ -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.5 → llm_dialog_manager-0.4.6}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "llm_dialog_manager"
-version = "0.4.5"
+version = "0.4.6"
 description = "A Python package for managing LLM chat conversation history"
 readme = "README.md"
 classifiers = [ "Development Status :: 3 - Alpha", "Intended Audience :: Developers", "License :: OSI Approved :: MIT License", "Operating System :: OS Independent", "Programming Language :: Python :: 3.8", "Programming Language :: Python :: 3.9", "Programming Language :: Python :: 3.10", "Topic :: Software Development :: Libraries :: Python Modules", "Topic :: Scientific/Engineering :: Artificial Intelligence",]