nora-lib-impl 1.8.0__py3-none-any.whl

nora_lib/impl/context/agent_context.py

@@ -0,0 +1,50 @@
+from typing import Optional
+from uuid import UUID
+
+from nora_lib.impl.interactions.models import Surface
+from pydantic import BaseModel
+
+from nora_lib.serializers import UuidWithSerializer
+
+
+class MessageAgentContext(BaseModel):
+    """
+    Identifiers for the triggering message
+    """
+
+    message_id: str
+    thread_id: str
+    channel_id: str
+    # Pydantic models don't actually need this, but some codebases pass model_dump() objects
+    # to other JSON serializers (e.g. FastAPI) which don't handle UUIDs and other python built-ins.
+    # So we use this special UUID wrapper for convenience.
+    actor_id: UuidWithSerializer
+    surface: Surface
+
+
+class PubsubAgentContext(BaseModel):
+    """
+    The pubsub namespace in which the Handler is running
+    """
+
+    base_url: str
+    namespace: str
+
+
+class ToolConfigAgentContext(BaseModel):
+    """
+    The name of the tool config being used by the handler (dev, prod, demo, etc.)
+    """
+
+    env: str
+
+
+class AgentContext(BaseModel):
+    """
+    Information that needs to be passed from the Handler to tool agents
+    """
+
+    message: MessageAgentContext
+    pubsub: PubsubAgentContext
+    tool_config: ToolConfigAgentContext
+    step_id: Optional[str] = None

nora_lib/impl/context/context_service.py

@@ -0,0 +1,33 @@
+from typing import Optional
+
+from nora_lib.impl.interactions.interactions_service import InteractionsService
+from nora_lib.impl.interactions.models import ReturnedMessage
+
+
+class ContextService:
+    """
+    Save and retrieve task agent context from interaction store
+    """
+
+    def __init__(
+        self,
+        agent_actor_id: str,  # uuid representing this agent in interaction store
+        interactions_base_url: str,
+        interactions_bearer_token: Optional[str],
+        timeout: int = 30,
+    ):
+        # If no config is provided, load the configuration based on the environment
+        self.interactions_service = self._get_interactions_service(
+            interactions_base_url, interactions_bearer_token, timeout
+        )
+        self.agent_actor_id = agent_actor_id
+
+    def _get_interactions_service(self, url, token, timeout) -> InteractionsService:
+        return InteractionsService(url, timeout, token)
+
+    def get_message(self, message_id: str) -> str:
+        message: ReturnedMessage = self.interactions_service.get_message(message_id)
+        if message.annotated_text:
+            return message.annotated_text
+        else:
+            return message.text

nora_lib/impl/context/models.py

@@ -0,0 +1,11 @@
+from typing import Optional
+from pydantic import BaseModel, Field
+
+
+class WrappedTaskObject(BaseModel):
+    """Encloses request or response object with additional metadata"""
+
+    message_id: str = Field(
+        description="id of originating message; key for istore retrieval"
+    )
+    data: dict = Field(description="Tool-defined request or response")