nora-lib-impl 1.0.0__tar.gz

nora_lib_impl-1.0.0/PKG-INFO

@@ -0,0 +1,20 @@
+Metadata-Version: 2.2
+Name: nora_lib-impl
+Version: 1.0.0
+Summary: For making and coordinating agents and tools
+Home-page: https://github.com/allenai/nora_lib/impl
+Requires-Python: >=3.9
+Requires-Dist: pydantic<3,>=2
+Requires-Dist: requests
+Requires-Dist: boto3
+Requires-Dist: aws_requests_auth
+Provides-Extra: dev
+Requires-Dist: mypy; extra == "dev"
+Requires-Dist: pytest; extra == "dev"
+Requires-Dist: black; extra == "dev"
+Requires-Dist: types-requests; extra == "dev"
+Dynamic: home-page
+Dynamic: provides-extra
+Dynamic: requires-dist
+Dynamic: requires-python
+Dynamic: summary

nora_lib_impl-1.0.0/README.md

@@ -0,0 +1,14 @@
+#nora_lib impl
+
+Contains implementations of `nora_lib` interfaces using the Nora platform
+
+Publishes the `nora_lib-impl` package
+
+# Development
+
+Verify changes using
+
+```
+make verify
+```
+

nora_lib_impl-1.0.0/pyproject.toml

@@ -0,0 +1,3 @@
+[build-system]
+requires = ["setuptools", "wheel"]
+build-backend = "setuptools.build_meta"

nora_lib_impl-1.0.0/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

nora_lib_impl-1.0.0/setup.py

@@ -0,0 +1,26 @@
+import os
+import setuptools
+
+runtime_requirements = ["pydantic>=2,<3", "requests", "boto3", "aws_requests_auth"]
+
+# For running tests, linting, etc
+dev_requirements = ["mypy", "pytest", "black", "types-requests"]
+
+version = os.environ["NORA_LIB_VERSION"]
+
+setuptools.setup(
+    name="nora_lib-impl",
+    version=version,
+    description="For making and coordinating agents and tools",
+    url="https://github.com/allenai/nora_lib/impl",
+    package_dir={"": "src"},
+    packages=setuptools.find_packages(where="src", include=["nora_lib*"],),
+    install_requires=runtime_requirements,
+    package_data={
+        "nora_lib": ["py.typed"],
+    },
+    extras_require={
+        "dev": dev_requirements,
+    },
+    python_requires=">=3.9",
+)

nora_lib_impl-1.0.0/src/nora_lib/impl/context/agent_context.py

@@ -0,0 +1,43 @@
+from typing import Optional
+
+from nora_lib.impl.interactions.models import Surface
+from pydantic import BaseModel
+
+
+class MessageAgentContext(BaseModel):
+    """
+    Identifiers for the triggering user message
+    """
+
+    message_id: str
+    thread_id: str
+    channel_id: str
+    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-1.0.0/src/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-1.0.0/src/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")