pop-python 1.0.2__tar.gz → 1.0.4__tar.gz

{pop_python-1.0.2 → pop_python-1.0.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pop-python
-Version: 1.0.2
+Version: 1.0.4
 Summary: Prompt Oriented Programming (POP): reusable, composable prompt functions for LLMs.
 Home-page: https://github.com/sgt1796/POP
 Author: Guotai Shen

{pop_python-1.0.2 → pop_python-1.0.4}/POP/LLMClient.py

@@ -84,11 +84,11 @@ class OpenAIClient(LLMClient):
             else:
                 request_payload["response_format"] = {"type": "json_schema", "json_schema": fmt}
 
-        # Handle function tools
+        # Handle function tools (already OpenAI-style)
         tools = kwargs.get("tools", None)
         if tools:
-            request_payload["tools"] = [{"type": "function", "function": tool} for tool in tools]
-            request_payload["tool_choice"] = "auto"
+            request_payload["tools"] = tools
+            request_payload["tool_choice"] = kwargs.get("tool_choice", "auto")
 
         # Temporary patch for models not supporting system roles
         if model == "o1-mini" and request_payload["messages"] and request_payload["messages"][0]["role"] == "system":
@@ -140,6 +140,12 @@ class DeepseekClient(LLMClient):
             role = msg.get("role", "user")
             request_payload["messages"].append({"role": role, "content": content})
 
+        # tools: OpenAI-style list
+        tools = kwargs.get("tools", None)
+        if tools:
+            request_payload["tools"] = tools
+            request_payload["tool_choice"] = kwargs.get("tool_choice", "auto")
+
         # Execute request
         try:
             response = self.client.chat.completions.create(**request_payload)
@@ -291,7 +297,8 @@ class DoubaoClient(LLMClient):
         # Tools (function calling)
         tools = kwargs.get("tools")
         if tools:
-            raise NotImplementedError("DoubaoClient does not support tools yet.")
+            payload["tools"] = tools
+            payload["tool_choice"] = kwargs.get("tool_choice", "auto")
         try:
             response = self.client.chat.completions.create(**payload)
         except Exception as e:
@@ -400,4 +407,4 @@ class OllamaClient(LLMClient):
         class Response:
             def __init__(self, content): self.choices = [Choice(Message(content))]
 
-        return Response(content)
+        return Response(content)

{pop_python-1.0.2 → pop_python-1.0.4}/POP/POP.py

@@ -3,7 +3,7 @@ import json
 import requests
 from dotenv import load_dotenv
 from os import getenv, path
-from LLMClient import LLMClient, OpenAIClient, GeminiClient, DeepseekClient, LocalPyTorchClient, DoubaoClient, OllamaClient
+from .LLMClient import LLMClient, OpenAIClient, GeminiClient, DeepseekClient, LocalPyTorchClient, DoubaoClient, OllamaClient
 
 # Load environment variables
 load_dotenv()
@@ -70,6 +70,7 @@ class PromptFunction:
             - sys: Additional system instructions.
             - fmt: Response format/schema.
             - tools: List of function tools to use (for function calling).
+            - tool_choice
             - temp: Temperature.
             - ADD_BEFORE: Text to prepend.
             - ADD_AFTER: Text to append.
@@ -87,6 +88,9 @@ class PromptFunction:
         tools = kwargs.pop("tools", None)
         temp = kwargs.pop("temp", self.temperature)
         images = kwargs.pop("images", None)
+        tool_choice = kwargs.pop("tool_choice", None)
+        if tools and not tool_choice:
+            tool_choice = "auto"
 
         # Prepare the prompt with dynamic injections.
         formatted_prompt = self._prepare_prompt(*args, **kwargs)
@@ -105,15 +109,19 @@ class PromptFunction:
         messages = [system_message, user_message]
 
         try:
-            # Call the LLM client.
-            raw_response = self.client.chat_completion(
-                messages=messages, 
-                model=model, 
-                temperature=temp, 
-                response_format=fmt,
-                tools=tools,
-                images=images
-            )
+            # Call the LLM client. Always include tool_choice when tools are provided.
+            call_kwargs = {
+                "messages": messages,
+                "model": model,
+                "temperature": temp,
+                "response_format": fmt,
+                "tools": tools,
+                "images": images,
+                # Always include tool_choice key so callers can assert on it
+                "tool_choice": tool_choice,
+            }
+
+            raw_response = self.client.chat_completion(**call_kwargs)
         except Exception as e:
             verbose = True
             if verbose:

{pop_python-1.0.2 → pop_python-1.0.4}/pop_python.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pop-python
-Version: 1.0.2
+Version: 1.0.4
 Summary: Prompt Oriented Programming (POP): reusable, composable prompt functions for LLMs.
 Home-page: https://github.com/sgt1796/POP
 Author: Guotai Shen

{pop_python-1.0.2 → pop_python-1.0.4}/setup.py

@@ -7,7 +7,7 @@ long_description = (this_dir / "README.md").read_text(encoding="utf-8")
 setup(
     # PyPI project name
     name="pop-python",
-    version="1.0.2",  # update as needed
+    version="1.0.4",  # update as needed
 
     author="Guotai Shen",
     author_email="sgt1796@gmail.com",