lionagi 0.12.7__py3-none-any.whl → 0.12.8__py3-none-any.whl

lionagi/service/connections/providers/claude_code_.py

@@ -83,11 +83,11 @@ ENDPOINT_CONFIG = EndpointConfig(
     endpoint="query",
     api_key="dummy",
     request_options=ClaudeCodeRequest,
+    timeout=3000,
 )
 
 
 class ClaudeCodeEndpoint(Endpoint):
-
     def __init__(self, config=ENDPOINT_CONFIG, **kwargs):
         super().__init__(config=config, **kwargs)
 
@@ -146,95 +146,51 @@ class ClaudeCodeEndpoint(Endpoint):
         When Claude Code uses tools, the ResultMessage.result may be None.
         In that case, we need to look at the tool results in UserMessages.
         """
-        result_message = None
-        model = "claude-code"
-        assistant_text_content = []
-        tool_results = []
+        results = {
+            "session_id": None,
+            "model": "claude-code",
+            "result": "",
+            "tool_results": [],
+            "is_error": False,
+            "num_turns": None,
+            "cost_usd": None,
+            "total_cost_usd": None,
+            "usage": {
+                "prompt_tokens": 0,
+                "completion_tokens": 0,
+                "total_tokens": 0,
+            },
+        }
+
+        from claude_code_sdk import types
 
-        # Process all messages
         for response in responses:
-            class_name = response.__class__.__name__
-
-            if class_name == "SystemMessage" and hasattr(response, "data"):
-                model = response.data.get("model", "claude-code")
-
-            elif class_name == "AssistantMessage":
-                # Extract text content from assistant messages
-                if hasattr(response, "content") and response.content:
-                    for block in response.content:
-                        if hasattr(block, "text"):
-                            assistant_text_content.append(block.text)
-                        elif isinstance(block, dict) and "text" in block:
-                            assistant_text_content.append(block["text"])
-
-            elif class_name == "UserMessage":
-                # Extract tool results from user messages
-                if hasattr(response, "content") and response.content:
-                    for item in response.content:
-                        if (
-                            isinstance(item, dict)
-                            and item.get("type") == "tool_result"
-                        ):
-                            tool_results.append(item.get("content", ""))
-
-            elif class_name == "ResultMessage":
-                result_message = response
-
-        # Determine the final content
-        final_content = ""
-        if (
-            result_message
-            and hasattr(result_message, "result")
-            and result_message.result
-        ):
-            # Use ResultMessage.result if available
-            final_content = result_message.result
-        elif assistant_text_content:
-            # Use assistant text content if available
-            final_content = "\n".join(assistant_text_content)
-        elif tool_results:
-            # If only tool results are available, use a generic summary
-            # (Claude Code typically provides its own summary after tool use)
-            final_content = (
-                "I've completed the requested task using the available tools."
-            )
-
-        # Build the clean chat completions response
-        result = {
-            "model": model,
-            "choices": [
-                {
-                    "index": 0,
-                    "message": {"role": "assistant", "content": final_content},
-                    "finish_reason": (
-                        "stop"
-                        if not (
-                            result_message
-                            and hasattr(result_message, "is_error")
-                            and result_message.is_error
+            if isinstance(response, types.SystemMessage):
+                results["session_id"] = response.data.get("session_id")
+                results["model"] = response.data.get("model", "claude-code")
+            if isinstance(response, types.AssistantMessage):
+                for block in response.content:
+                    if isinstance(block, types.TextBlock):
+                        results["result"] += block.text.strip() + "\n"
+                    if isinstance(block, types.ToolResultBlock):
+                        results["tool_results"].append(
+                            {
+                                "tool_use_id": block.tool_use_id,
+                                "content": block.content,
+                                "is_error": block.is_error,
+                            }
                         )
-                        else "error"
-                    ),
-                }
-            ],
-        }
-
-        # Add usage information if available
-        if result_message and hasattr(result_message, "usage"):
-            result["usage"] = result_message.usage
-
-        # Add only essential Claude Code metadata
-        if result_message:
-            if hasattr(result_message, "cost_usd"):
-                result["usage"]["cost_usd"] = result_message.cost_usd
-            if hasattr(result_message, "session_id"):
-                result["session_id"] = result_message.session_id
-            if hasattr(result_message, "is_error"):
-                result["is_error"] = result_message.is_error
-            if hasattr(result_message, "num_turns"):
-                result["num_turns"] = result_message.num_turns
-
-        return result
+            if isinstance(response, types.ResultMessage):
+                results["result"] += response.result.strip() or ""
+                results["usage"] = response.usage
+                results["cost_usd"] = response.cost_usd
+                results["is_error"] = response.is_error
+                results["total_cost_usd"] = response.total_cost_usd
+                results["num_turns"] = response.num_turns
+                results["duration_ms"] = response.duration_ms
+                results["duration_api_ms"] = response.duration_api_ms
+
+        return results
 
     async def _call(
         self,

lionagi/service/connections/providers/oai_.py

@@ -98,10 +98,21 @@ GROQ_CHAT_ENDPOINT_CONFIG = EndpointConfig(
 
 
 REASONING_MODELS = (
-    "o3-mini-2025-01-31",
-    "o3-mini",
     "o1",
     "o1-2024-12-17",
+    "o1-preview-2024-09-12",
+    "o1-pro",
+    "o1-pro-2025-03-19",
+    "o3-pro",
+    "o3-pro-2025-06-10",
+    "o3",
+    "o3-2025-04-16",
+    "o4-mini",
+    "o4-mini-2025-04-16",
+    "o3-mini",
+    "o3-mini-2025-01-31",
+    "o1-mini",
+    "o1-mini-2024-09-12",
 )
 
 REASONING_NOT_SUPPORT_PARAMS = (

lionagi/version.py

@@ -1 +1 @@
-__version__ = "0.12.7"
+__version__ = "0.12.8"

{lionagi-0.12.7.dist-info → lionagi-0.12.8.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lionagi
-Version: 0.12.7
+Version: 0.12.8
 Summary: An Intelligence Operating System.
 Author-email: HaiyangLi <quantocean.li@gmail.com>, Liangbingyan Luo <llby_luo@outlook.com>
 License:            Apache License

{lionagi-0.12.7.dist-info → lionagi-0.12.8.dist-info}/RECORD

@@ -6,7 +6,7 @@ lionagi/config.py,sha256=dAhDFKtaaSfn6WT9dwX9Vd4TWWs6-Su1FgYIrFgYcgc,3709
 lionagi/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 lionagi/settings.py,sha256=W52mM34E6jXF3GyqCFzVREKZrmnUqtZm_BVDsUiDI_s,1627
 lionagi/utils.py,sha256=uLTJKl7aTnFXV6ehA6zwiwEB7G2nQYKsO2pZ6mqFzUk,78908
-lionagi/version.py,sha256=WvcWmk-c-I-XXiq6oM9Y1VYwNWFe64tXpnG_RI47HhE,23
+lionagi/version.py,sha256=F7xxYe0dXryqS1cGEXFikx8AI7-UsZzdi89hJdyx-b0,23
 lionagi/fields/__init__.py,sha256=8oU7Vfk-fKiULFKqhM6VpJMqdZcVXPTM7twVfNDN_SQ,603
 lionagi/fields/action.py,sha256=iWSApCM77jS0Oc28lb7G601Etkp-yjx5U1hfI_FQgfA,5792
 lionagi/fields/base.py,sha256=5CJc7j8kTTWzXwpYzkSAFzx4BglABfx3AElIATKB7bg,3857
@@ -177,9 +177,9 @@ lionagi/service/connections/header_factory.py,sha256=22sG4ian3MiNklF6SdQqkEYgtWK
 lionagi/service/connections/match_endpoint.py,sha256=mEZPDkK1qtvjTGN9-PZsK7w_yB7642nZiJsb0l5QUx4,1827
 lionagi/service/connections/providers/__init__.py,sha256=3lzOakDoBWmMaNnT2g-YwktPKa_Wme4lnPRSmOQfayY,105
 lionagi/service/connections/providers/anthropic_.py,sha256=SUPnw2UqjY5wuHXLHas6snMTzhQ-UuixvPYbkVnXn34,3083
-lionagi/service/connections/providers/claude_code_.py,sha256=1kCLHjc-bqjCsG6HieaJQWy6SCV6PzBZGjTneLPhCY0,8845
+lionagi/service/connections/providers/claude_code_.py,sha256=o7tfhCcOxkOKe3dwdzIme6ov0B3Os1Xiu9uewIw2pg0,7182
 lionagi/service/connections/providers/exa_.py,sha256=GGWaD9jd5gKM257OfUaIBBKIqR1NrNcBE67p_7JbK7g,938
-lionagi/service/connections/providers/oai_.py,sha256=rKZdGufX4vpj88rr98KqoOHzHa7HgH_pCVWjCqAq-XU,4971
+lionagi/service/connections/providers/oai_.py,sha256=hH5oEMvS9A0Kbvh-lSFm5dLiqPV2gfRra6zzDOq78P0,5191
 lionagi/service/connections/providers/ollama_.py,sha256=jdx6dGeChwVk5TFfFRbpnrpKzj8YQZw6D5iWJ6zYmfk,4096
 lionagi/service/connections/providers/perplexity_.py,sha256=9MH9YmMy9Jg7JDMJHQxxMYHyjJ4NP0OlN7sCuhla85I,917
 lionagi/service/third_party/README.md,sha256=qFjWnI8rmLivIyr6Tc-hRZh-rQwntROp76af4MBNJJc,2214
@@ -198,7 +198,7 @@ lionagi/tools/types.py,sha256=XtJLY0m-Yi_ZLWhm0KycayvqMCZd--HxfQ0x9vFUYDE,230
 lionagi/tools/file/__init__.py,sha256=5y5joOZzfFWERl75auAcNcKC3lImVJ5ZZGvvHZUFCJM,112
 lionagi/tools/file/reader.py,sha256=0TdnfVGVCKuM58MmGM-NyVjhU9BFoitkNYEepdc0z_Y,9529
 lionagi/tools/memory/tools.py,sha256=zTGBenVsF8Wuh303kWntmQSGlAFKonHNdh5ePuQ26KE,15948
-lionagi-0.12.7.dist-info/METADATA,sha256=6cpK8bMdHTYDG78vn9Gl09ENET3A4vYkfaot81uoAa0,20200
-lionagi-0.12.7.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-lionagi-0.12.7.dist-info/licenses/LICENSE,sha256=VXFWsdoN5AAknBCgFqQNgPWYx7OPp-PFEP961zGdOjc,11288
-lionagi-0.12.7.dist-info/RECORD,,
+lionagi-0.12.8.dist-info/METADATA,sha256=8cD7ID0yI0CK3JqF0MelRZPsodoqUS1GyQJsnW44yZs,20200
+lionagi-0.12.8.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+lionagi-0.12.8.dist-info/licenses/LICENSE,sha256=VXFWsdoN5AAknBCgFqQNgPWYx7OPp-PFEP961zGdOjc,11288
+lionagi-0.12.8.dist-info/RECORD,,