llm-dialog-manager 0.4.1__tar.gz → 0.4.2__tar.gz

{llm_dialog_manager-0.4.1 → llm_dialog_manager-0.4.2}/PKG-INFO

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

{llm_dialog_manager-0.4.1 → llm_dialog_manager-0.4.2}/llm_dialog_manager/__init__.py

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

{llm_dialog_manager-0.4.1 → llm_dialog_manager-0.4.2}/llm_dialog_manager/agent.py

@@ -74,7 +74,10 @@ def completion(model: str, messages: List[Dict[str, Union[str, List[Union[str, I
     """
     try:
         service = ""
-        if "claude" in model:
+        if "openai" in model:
+            service = "openai"
+            model
+        elif "claude" in model:
             service = "anthropic"
         elif "gemini" in model:
             service = "gemini"
@@ -92,7 +95,7 @@ def completion(model: str, messages: List[Dict[str, Union[str, List[Union[str, I
 
         def format_messages_for_api(model, messages):
             """Convert ChatHistory messages to the format required by the specific API."""
-            if "claude" in model:
+            if "claude" in model and "openai" not in model:
                 formatted = []
                 system_msg = ""
                 if messages and messages[0]["role"] == "system":
@@ -141,7 +144,7 @@ def completion(model: str, messages: List[Dict[str, Union[str, List[Union[str, I
                         formatted.append({"role": msg["role"], "content": combined_content})
                 return system_msg, formatted
 
-            elif "gemini" in model or "gpt" in model or "grok" in model:
+            elif ("gemini" in model or "gpt" in model or "grok" in model) and "openai" not in model:
                 formatted = []
                 for msg in messages:
                     content = msg["content"]
@@ -191,7 +194,7 @@ def completion(model: str, messages: List[Dict[str, Union[str, List[Union[str, I
 
         system_msg, formatted_messages = format_messages_for_api(model, messages.copy())
 
-        if "claude" in model:
+        if "claude" in model and "openai" not in model:
             # Check for Vertex configuration
             vertex_project_id = os.getenv('VERTEX_PROJECT_ID')
             vertex_region = os.getenv('VERTEX_REGION')
@@ -232,7 +235,7 @@ def completion(model: str, messages: List[Dict[str, Union[str, List[Union[str, I
 
             return response.completion
 
-        elif "gemini" in model:
+        elif "gemini" in model and "openai" not in model:
             try:
                 # First try OpenAI-style API
                 client = openai.OpenAI(
@@ -284,7 +287,7 @@ def completion(model: str, messages: List[Dict[str, Union[str, List[Union[str, I
 
                 return response.text
 
-        elif "grok" in model:
+        elif "grok" in model and "openai" not in model:
             # Randomly choose between OpenAI and Anthropic SDK
             use_anthropic = random.choice([True, False])
 
@@ -326,6 +329,8 @@ def completion(model: str, messages: List[Dict[str, Union[str, List[Union[str, I
                 return response.choices[0].message.content
 
         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"}

{llm_dialog_manager-0.4.1 → llm_dialog_manager-0.4.2}/llm_dialog_manager.egg-info/PKG-INFO

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

{llm_dialog_manager-0.4.1 → llm_dialog_manager-0.4.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "llm_dialog_manager"
-version = "0.4.1"
+version = "0.4.2"
 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",]