llm-dialog-manager 0.4.5__py3-none-any.whl → 0.4.7__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.7" 

llm_dialog_manager/agent.py

@@ -393,17 +393,27 @@ def completion(model: str, messages: List[Dict[str, Union[str, List[Union[str, I
         else:  # OpenAI models
             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
-            )
+            
+            # Initialize OpenAI client with only supported parameters
+            client_kwargs = {"api_key": api_key}
+            if base_url:
+                client_kwargs["base_url"] = base_url
+            client = openai.OpenAI(**client_kwargs)
+            
+            # 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.7.dist-info}/METADATA

@@ -1,6 +1,6 @@
-Metadata-Version: 2.2
+Metadata-Version: 2.4
 Name: llm_dialog_manager
-Version: 0.4.5
+Version: 0.4.7
 Summary: A Python package for managing LLM chat conversation history
 Author-email: xihajun <work@2333.fun>
 License: MIT
@@ -44,6 +44,7 @@ Requires-Dist: pytest-asyncio>=0.21.1; extra == "all"
 Requires-Dist: pytest-cov>=4.1.0; extra == "all"
 Requires-Dist: black>=23.9.1; extra == "all"
 Requires-Dist: isort>=5.12.0; extra == "all"
+Dynamic: license-file
 
 # LLM Dialog Manager
 

llm_dialog_manager-0.4.7.dist-info/RECORD

@@ -0,0 +1,9 @@
+llm_dialog_manager/__init__.py,sha256=uWTAXzyON6m5vqj_MS8_Pe7hga5TZ2YDhkMh0Z-FDew,86
+llm_dialog_manager/agent.py,sha256=LGoX6erGcn9eaM4wRGfs0N5zS8Lop_57nPKzzBWrtk8,27871
+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.7.dist-info/licenses/LICENSE,sha256=vWGbYgGuWpWrXL8-xi6pNcX5UzD6pWoIAZmcetyfbus,1064
+llm_dialog_manager-0.4.7.dist-info/METADATA,sha256=mRNR-f-rRqDR_IFoFJe2oNcTVE27Ah6AN8ENGQwfzVg,4216
+llm_dialog_manager-0.4.7.dist-info/WHEEL,sha256=1tXe9gY0PYatrMPMDd6jXqjfpz_B-Wqm32CPfRC58XU,91
+llm_dialog_manager-0.4.7.dist-info/top_level.txt,sha256=u2EQEXW0NGAt0AAHT7jx1odXZ4rZfjcgbmJhvKFuMkI,19
+llm_dialog_manager-0.4.7.dist-info/RECORD,,

{llm_dialog_manager-0.4.5.dist-info → llm_dialog_manager-0.4.7.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (75.8.0)
+Generator: setuptools (77.0.3)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

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,,