copilotkit 0.1.43a0__tar.gz → 0.2.0a0__tar.gz

copilotkit-0.2.0a0/PKG-INFO

@@ -0,0 +1,219 @@
+Metadata-Version: 2.3
+Name: copilotkit
+Version: 0.2.0a0
+Summary: CopilotKit python SDK
+License: MIT
+Keywords: copilot,copilotkit,langgraph,langchain,ai,langsmith,langserve
+Author: Markus Ecker
+Author-email: markus.ecker@gmail.com
+Requires-Python: >=3.10,<3.13
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Provides-Extra: crewai
+Requires-Dist: crewai (==0.114.0) ; extra == "crewai"
+Requires-Dist: fastapi (>=0.115.0,<0.116.0)
+Requires-Dist: langchain (>=0.3.3,<0.4.0)
+Requires-Dist: langgraph (>=0.2.50)
+Requires-Dist: partialjson (>=0.0.8,<0.0.9)
+Requires-Dist: toml (>=0.10.2,<0.11.0)
+Project-URL: Homepage, https://copilotkit.ai
+Project-URL: Repository, https://github.com/CopilotKit/CopilotKit/tree/main/sdk-python
+Description-Content-Type: text/markdown
+
+# CopilotKit Python SDK
+
+[![PyPI version](https://badge.fury.io/py/copilotkit.svg)](https://badge.fury.io/py/copilotkit)
+[![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](https://opensource.org/licenses/MIT)
+
+The official Python SDK for CopilotKit - build AI copilots and agents into your applications.
+
+## Features
+
+- 🚀 Integration with LangGraph and LangChain
+- 🔄 Support for stateful conversations
+- 🔌 FastAPI integration for serving endpoints
+- 🤝 Optional CrewAI integration (requires CrewAI 0.114.0+)
+
+## Version 0.2.0 Breaking Changes
+
+⚠️ As of version 0.2.0, CopilotKit Python SDK now requires CrewAI 0.114.0 or higher.
+If you're using CrewAI with CopilotKit, you must update your CrewAI version.
+
+## Installation
+
+```bash
+pip install copilotkit
+```
+
+With CrewAI support:
+
+```bash
+pip install "copilotkit[crewai]"
+```
+
+## Quick Start
+
+```python
+from copilotkit import CopilotKitRemoteEndpoint, Action
+from copilotkit.integrations.fastapi import add_fastapi_endpoint
+from fastapi import FastAPI
+
+# Define an action handler function
+def greet_user_handler(name: str):
+    return f"Hello, {name}!"
+
+# Initialize the SDK endpoint
+endpoint = CopilotKitRemoteEndpoint(
+    actions=[
+        Action(
+            name="greet_user",
+            handler=greet_user_handler,
+            description="Greet the user",
+            parameters=[
+                {
+                    "name": "name",
+                    "type": "string",
+                    "description": "The name of the user"
+                }
+            ]
+        )
+    ]
+)
+
+# Create FastAPI app and add CopilotKit endpoint
+app = FastAPI()
+add_fastapi_endpoint(app, endpoint, "/copilotkit")
+```
+
+## LangGraph Integration
+
+CopilotKit provides seamless integration with LangGraph for building sophisticated agent workflows.
+
+### Using LangGraphAgent
+
+```python
+from copilotkit import CopilotKitRemoteEndpoint, LangGraphAgent
+from langgraph.graph import StateGraph, MessagesState
+
+# Create your LangGraph
+def create_graph():
+    workflow = StateGraph(MessagesState)
+    # ... define your graph nodes and edges
+    return workflow.compile()
+
+# Initialize the SDK with a LangGraph agent
+endpoint = CopilotKitRemoteEndpoint(
+    agents=[
+        LangGraphAgent(
+            name="email_agent",
+            description="This agent sends emails",
+            graph=create_graph(),
+        )
+    ]
+)
+```
+
+### Customizing LangGraph Behavior
+
+CopilotKit provides utilities to customize how your LangGraph agents work:
+
+```python
+from copilotkit.langgraph import copilotkit_customize_config, copilotkit_emit_state
+
+# Customize configuration
+config = copilotkit_customize_config(
+    base_config=None,  # or your existing config
+    emit_messages=True,
+    emit_tool_calls=True,
+    emit_intermediate_state=[
+        {
+            "state_key": "progress",
+            "tool": "ProgressTool",
+            "tool_argument": "steps"
+        },
+    ]
+)
+
+# Emit intermediate state
+async def long_running_node(state):
+    for i in range(10):
+        await some_operation(i)
+        await copilotkit_emit_state(config, {"progress": i})
+    return state
+```
+
+## CrewAI Integration
+
+CopilotKit provides integration with CrewAI for building multi-agent systems.
+
+### Intercepting CopilotKit Actions in CrewAI Flows
+
+When building flows that use CopilotKit actions, you need to properly intercept any tool calls that should be handled by the frontend:
+
+```python
+from copilotkit.crewai import create_copilotkit_tool_handlers, check_for_intercepted_actions
+
+@router(start_flow)
+async def chat(self):
+    # Your existing code...
+
+    # 1. Create wrapped tool handlers
+    wrapped_handlers = create_copilotkit_tool_handlers(
+        original_handlers=tool_handlers,  # Your original tool handlers
+        copilotkit_actions=self.state.copilotkit.actions,  # From CopilotKitState
+        state=self.state  # Flow state for flagging
+    )
+
+    # 2. Use wrapped handlers with LLM call
+    response = llm.call(
+        messages=[...],
+        tools=[
+            *self.state.copilotkit.actions,  # Include CopilotKit actions
+            YOUR_OTHER_TOOLS...
+        ],
+        available_functions=wrapped_handlers  # Use wrapped handlers
+    )
+
+    # 3. Create response message as usual
+    message = {"role": "assistant", "content": response}
+    self.state.messages.append(message)
+
+    # 4. Check if a CopilotKit action was intercepted
+    if check_for_intercepted_actions(self.state):
+        # Return to frontend to handle the action
+        return "route_end"
+
+    # Continue with normal flow...
+```
+
+This pattern ensures that:
+
+1. CopilotKit actions are properly identified
+2. The frontend gets control when needed
+3. Other tools continue to work normally
+
+## Documentation
+
+For detailed documentation and examples, visit [copilotkit.ai](https://copilotkit.ai)
+
+## Contributing
+
+We welcome contributions! Please see our [Contributing Guidelines](https://github.com/CopilotKit/CopilotKit/blob/main/CONTRIBUTING.md) for details.
+
+## License
+
+This project is licensed under the MIT License - see the [LICENSE](https://github.com/CopilotKit/CopilotKit/blob/main/LICENSE) file for details.
+
+## Support
+
+- 📚 [Documentation](https://docs.copilotkit.ai)
+- 💬 [Discord Community](https://discord.gg/6dffbvGU)
+- 🐛 [Issue Tracker](https://github.com/CopilotKit/CopilotKit/issues)
+
+---
+
+Built with ❤️ by the CopilotKit team
+

copilotkit-0.2.0a0/README.md

@@ -0,0 +1,193 @@
+# CopilotKit Python SDK
+
+[![PyPI version](https://badge.fury.io/py/copilotkit.svg)](https://badge.fury.io/py/copilotkit)
+[![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](https://opensource.org/licenses/MIT)
+
+The official Python SDK for CopilotKit - build AI copilots and agents into your applications.
+
+## Features
+
+- 🚀 Integration with LangGraph and LangChain
+- 🔄 Support for stateful conversations
+- 🔌 FastAPI integration for serving endpoints
+- 🤝 Optional CrewAI integration (requires CrewAI 0.114.0+)
+
+## Version 0.2.0 Breaking Changes
+
+⚠️ As of version 0.2.0, CopilotKit Python SDK now requires CrewAI 0.114.0 or higher.
+If you're using CrewAI with CopilotKit, you must update your CrewAI version.
+
+## Installation
+
+```bash
+pip install copilotkit
+```
+
+With CrewAI support:
+
+```bash
+pip install "copilotkit[crewai]"
+```
+
+## Quick Start
+
+```python
+from copilotkit import CopilotKitRemoteEndpoint, Action
+from copilotkit.integrations.fastapi import add_fastapi_endpoint
+from fastapi import FastAPI
+
+# Define an action handler function
+def greet_user_handler(name: str):
+    return f"Hello, {name}!"
+
+# Initialize the SDK endpoint
+endpoint = CopilotKitRemoteEndpoint(
+    actions=[
+        Action(
+            name="greet_user",
+            handler=greet_user_handler,
+            description="Greet the user",
+            parameters=[
+                {
+                    "name": "name",
+                    "type": "string",
+                    "description": "The name of the user"
+                }
+            ]
+        )
+    ]
+)
+
+# Create FastAPI app and add CopilotKit endpoint
+app = FastAPI()
+add_fastapi_endpoint(app, endpoint, "/copilotkit")
+```
+
+## LangGraph Integration
+
+CopilotKit provides seamless integration with LangGraph for building sophisticated agent workflows.
+
+### Using LangGraphAgent
+
+```python
+from copilotkit import CopilotKitRemoteEndpoint, LangGraphAgent
+from langgraph.graph import StateGraph, MessagesState
+
+# Create your LangGraph
+def create_graph():
+    workflow = StateGraph(MessagesState)
+    # ... define your graph nodes and edges
+    return workflow.compile()
+
+# Initialize the SDK with a LangGraph agent
+endpoint = CopilotKitRemoteEndpoint(
+    agents=[
+        LangGraphAgent(
+            name="email_agent",
+            description="This agent sends emails",
+            graph=create_graph(),
+        )
+    ]
+)
+```
+
+### Customizing LangGraph Behavior
+
+CopilotKit provides utilities to customize how your LangGraph agents work:
+
+```python
+from copilotkit.langgraph import copilotkit_customize_config, copilotkit_emit_state
+
+# Customize configuration
+config = copilotkit_customize_config(
+    base_config=None,  # or your existing config
+    emit_messages=True,
+    emit_tool_calls=True,
+    emit_intermediate_state=[
+        {
+            "state_key": "progress",
+            "tool": "ProgressTool",
+            "tool_argument": "steps"
+        },
+    ]
+)
+
+# Emit intermediate state
+async def long_running_node(state):
+    for i in range(10):
+        await some_operation(i)
+        await copilotkit_emit_state(config, {"progress": i})
+    return state
+```
+
+## CrewAI Integration
+
+CopilotKit provides integration with CrewAI for building multi-agent systems.
+
+### Intercepting CopilotKit Actions in CrewAI Flows
+
+When building flows that use CopilotKit actions, you need to properly intercept any tool calls that should be handled by the frontend:
+
+```python
+from copilotkit.crewai import create_copilotkit_tool_handlers, check_for_intercepted_actions
+
+@router(start_flow)
+async def chat(self):
+    # Your existing code...
+
+    # 1. Create wrapped tool handlers
+    wrapped_handlers = create_copilotkit_tool_handlers(
+        original_handlers=tool_handlers,  # Your original tool handlers
+        copilotkit_actions=self.state.copilotkit.actions,  # From CopilotKitState
+        state=self.state  # Flow state for flagging
+    )
+
+    # 2. Use wrapped handlers with LLM call
+    response = llm.call(
+        messages=[...],
+        tools=[
+            *self.state.copilotkit.actions,  # Include CopilotKit actions
+            YOUR_OTHER_TOOLS...
+        ],
+        available_functions=wrapped_handlers  # Use wrapped handlers
+    )
+
+    # 3. Create response message as usual
+    message = {"role": "assistant", "content": response}
+    self.state.messages.append(message)
+
+    # 4. Check if a CopilotKit action was intercepted
+    if check_for_intercepted_actions(self.state):
+        # Return to frontend to handle the action
+        return "route_end"
+
+    # Continue with normal flow...
+```
+
+This pattern ensures that:
+
+1. CopilotKit actions are properly identified
+2. The frontend gets control when needed
+3. Other tools continue to work normally
+
+## Documentation
+
+For detailed documentation and examples, visit [copilotkit.ai](https://copilotkit.ai)
+
+## Contributing
+
+We welcome contributions! Please see our [Contributing Guidelines](https://github.com/CopilotKit/CopilotKit/blob/main/CONTRIBUTING.md) for details.
+
+## License
+
+This project is licensed under the MIT License - see the [LICENSE](https://github.com/CopilotKit/CopilotKit/blob/main/LICENSE) file for details.
+
+## Support
+
+- 📚 [Documentation](https://docs.copilotkit.ai)
+- 💬 [Discord Community](https://discord.gg/6dffbvGU)
+- 🐛 [Issue Tracker](https://github.com/CopilotKit/CopilotKit/issues)
+
+---
+
+Built with ❤️ by the CopilotKit team

{copilotkit-0.1.43a0 → copilotkit-0.2.0a0}/copilotkit/crewai/__init__.py

@@ -12,6 +12,11 @@ from .crewai_sdk import (
     copilotkit_exit,
     copilotkit_predict_state,
 )
+from .utils import (
+    create_copilotkit_tool_handlers, 
+    check_for_intercepted_actions,
+    COPILOTKIT_ACTION_INTERCEPTED_MARKER
+)
 __all__ = [
     "CrewAIAgent",
     "CopilotKitProperties",
@@ -22,4 +27,7 @@ __all__ = [
     "copilotkit_stream",
     "copilotkit_exit",
     "copilotkit_predict_state",
+    "create_copilotkit_tool_handlers",
+    "check_for_intercepted_actions",
+    "COPILOTKIT_ACTION_INTERCEPTED_MARKER",
 ]

{copilotkit-0.1.43a0 → copilotkit-0.2.0a0}/copilotkit/crewai/crewai_agent.py

@@ -5,9 +5,9 @@ CrewAI Agent
 import uuid
 import json
 from copy import deepcopy
-from pydantic import BaseModel
 from typing import Optional, List, Callable
 from typing_extensions import TypedDict, NotRequired, Any, Dict, cast
+from pydantic import BaseModel
 from crewai import Crew, Flow
 from crewai.flow import start
 from crewai.cli.crew_chat import (
@@ -25,12 +25,13 @@ from copilotkit.protocol import (
   emit_runtime_events,
   agent_state_message,
 )
-from .crewai_sdk import (
+from copilotkit.crewai.crewai_sdk import (
   copilotkit_messages_to_crewai_flow,
   crewai_flow_messages_to_copilotkit,
   crewai_flow_async_runner,
   copilotkit_stream,
-  copilotkit_exit
+  copilotkit_exit,
+  logger
 )
 
 from copilotkit.runloop import copilotkit_run, CopilotKitRunExecution
@@ -296,14 +297,21 @@ class CrewAIAgent(Agent):
         if self.flow and self.flow._persistence: # pylint: disable=protected-access
             try:
                 stored_state = self.flow._persistence.load_state(thread_id) # pylint: disable=protected-access
+                messages = []
+                if "messages" in stored_state and stored_state["messages"]:
+                    try:
+                        messages = crewai_flow_messages_to_copilotkit(stored_state["messages"])
+                    except Exception as e: # pylint: disable=broad-except
+                        # If conversion fails, we'll return empty messages
+                        logger.warning(f"Failed to convert messages from stored state: {str(e)}")
                 return {
                     "threadId": thread_id,
                     "threadExists": True,
                     "state": stored_state,
-                    "messages": []
+                    "messages": messages
                 }
-            except: # pylint: disable=bare-except
-                pass
+            except Exception as e: # pylint: disable=broad-except
+                logger.warning(f"Failed to load state for thread {thread_id}: {str(e)}")
 
         return {
             "threadId": thread_id,