nora-lib 0.0.5.dev1__tar.gz → 0.0.5.dev2__tar.gz

{nora_lib-0.0.5.dev1 → nora_lib-0.0.5.dev2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: nora_lib
-Version: 0.0.5.dev1
+Version: 0.0.5.dev2
 Summary: For making and coordinating agents and tools
 Home-page: https://github.com/allenai/nora_lib
 Requires-Python: >=3.9

{nora_lib-0.0.5.dev1 → nora_lib-0.0.5.dev2}/nora_lib/interactions/interactions_service.py

@@ -4,6 +4,7 @@ import requests
 from typing import List, Optional
 
 from nora_lib.interactions.models import (
+    AnnotationBatch,
     Event,
     EventType,
     Message,
@@ -45,6 +46,17 @@ class InteractionsService:
         )
         response.raise_for_status()
 
+    def save_annotation(self, annotation: AnnotationBatch) -> None:
+        """Save an annotation to the Interactions API"""
+        annotation_url = f"{self.base_url}/interaction/v1/annotation"
+        response = requests.post(
+            annotation_url,
+            json=annotation.model_dump(),
+            headers=self.headers,
+            timeout=int(self.timeout),
+        )
+        response.raise_for_status()
+
     def get_message(self, message_id: str) -> ReturnedMessage:
         """Fetch a message from the Interactions API"""
         message_url = f"{self.base_url}/interaction/v1/search/message"

{nora_lib-0.0.5.dev1 → nora_lib-0.0.5.dev2}/nora_lib/interactions/models.py

@@ -25,6 +25,16 @@ class Annotation(BaseModel):
     attributes: Optional[Dict[str, str]] = None
 
 
+class AnnotationBatch(BaseModel):
+    actor_id: UUID
+    message_id: str
+    annotations: List[Annotation]
+
+    @field_serializer("actor_id")
+    def serialize_actor_id(self, actor_id: UUID):
+        return str(actor_id)
+
+
 class Message(BaseModel):
     message_id: str
     actor_id: UUID

{nora_lib-0.0.5.dev1 → nora_lib-0.0.5.dev2}/nora_lib.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: nora_lib
-Version: 0.0.5.dev1
+Version: 0.0.5.dev2
 Summary: For making and coordinating agents and tools
 Home-page: https://github.com/allenai/nora_lib
 Requires-Python: >=3.9

{nora_lib-0.0.5.dev1 → nora_lib-0.0.5.dev2}/setup.py

@@ -7,7 +7,7 @@ dev_requirements = ["mypy", "pytest", "black", "types-requests"]
 
 setuptools.setup(
     name="nora_lib",
-    version="0.0.5.dev1",
+    version="0.0.5.dev2",
     description="For making and coordinating agents and tools",
     url="https://github.com/allenai/nora_lib",
     packages=setuptools.find_packages(exclude=(["tests"])),