copilotkit 0.1.77__tar.gz → 0.1.78a0__tar.gz

{copilotkit-0.1.77 → copilotkit-0.1.78a0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: copilotkit
-Version: 0.1.77
+Version: 0.1.78a0
 Summary: CopilotKit python SDK
 License: MIT
 Keywords: copilot,copilotkit,langgraph,langchain,ai,langsmith,langserve

{copilotkit-0.1.77 → copilotkit-0.1.78a0}/copilotkit/copilotkit_lg_middleware.py

@@ -14,9 +14,10 @@ Example:
     )
 """
 
+import json
 from typing import Any, Callable, Awaitable, ClassVar, List
 
-from langchain_core.messages import AIMessage
+from langchain_core.messages import AIMessage, SystemMessage
 from langchain.agents.middleware import (
     AgentMiddleware,
     AgentState,
@@ -75,6 +76,100 @@ class CopilotKitMiddleware(AgentMiddleware[StateSchema, Any]):
 
         return await handler(request.override(tools=merged_tools))
 
+    # Inject app context before agent runs
+    def before_agent(
+            self,
+            state: StateSchema,
+            runtime: Runtime[Any],
+    ) -> dict[str, Any] | None:
+        messages = state.get("messages", [])
+
+        if not messages:
+            return None
+
+        # Get app context from state or runtime
+        copilotkit_state = state.get("copilotkit", {})
+        app_context = copilotkit_state.get("context") or getattr(runtime, "context", None)
+
+        # Check if app_context is missing or empty
+        if not app_context:
+            return None
+        if isinstance(app_context, str) and app_context.strip() == "":
+            return None
+        if isinstance(app_context, dict) and len(app_context) == 0:
+            return None
+
+        # Create the context content
+        if isinstance(app_context, str):
+            context_content = app_context
+        else:
+            context_content = json.dumps(app_context, indent=2)
+
+        context_message_content = f"App Context:\n{context_content}"
+        context_message_prefix = "App Context:\n"
+
+        # Helper to get message content as string
+        def get_content_string(msg: Any) -> str | None:
+            content = getattr(msg, "content", None)
+            if isinstance(content, str):
+                return content
+            if isinstance(content, list) and content and isinstance(content[0], dict):
+                return content[0].get("text")
+            return None
+
+        # Find the first system/developer message (not our context message)
+        # to determine where to insert our context message (right after it)
+        first_system_index = -1
+
+        for i, msg in enumerate(messages):
+            msg_type = getattr(msg, "type", None)
+            if msg_type in ("system", "developer"):
+                content = get_content_string(msg)
+                # Skip if this is our own context message
+                if content and content.startswith(context_message_prefix):
+                    continue
+                first_system_index = i
+                break
+
+        # Check if our context message already exists
+        existing_context_index = -1
+        for i, msg in enumerate(messages):
+            msg_type = getattr(msg, "type", None)
+            if msg_type in ("system", "developer"):
+                content = get_content_string(msg)
+                if content and content.startswith(context_message_prefix):
+                    existing_context_index = i
+                    break
+
+        # Create the context message
+        context_message = SystemMessage(content=context_message_content)
+
+        if existing_context_index != -1:
+            # Replace existing context message
+            updated_messages = list(messages)
+            updated_messages[existing_context_index] = context_message
+        else:
+            # Insert after the first system message, or at position 0 if no system message
+            insert_index = first_system_index + 1 if first_system_index != -1 else 0
+            updated_messages = [
+                *messages[:insert_index],
+                context_message,
+                *messages[insert_index:],
+            ]
+
+        return {
+            **state,
+            "messages": updated_messages,
+        }
+
+    async def abefore_agent(
+            self,
+            state: StateSchema,
+            runtime: Runtime[Any],
+    ) -> dict[str, Any] | None:
+        # Delegate to sync implementation
+        return self.before_agent(state, runtime)
+
     # Intercept frontend tool calls after model returns, before ToolNode executes
     def after_model(
             self,

{copilotkit-0.1.77 → copilotkit-0.1.78a0}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "copilotkit"
-version = "0.1.77"
+version = "0.1.78-alpha.0"
 description = "CopilotKit python SDK"
 authors = ["Markus Ecker <markus.ecker@gmail.com>"]
 license = "MIT"