nora-lib 0.0.7.dev1__tar.gz → 0.1.7__tar.gz

{nora_lib-0.0.7.dev1 → nora_lib-0.1.7}/PKG-INFO

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

{nora_lib-0.0.7.dev1 → nora_lib-0.1.7}/nora_lib/context/context_service.py

@@ -1,17 +1,7 @@
-from datetime import datetime, timezone
-from typing import List, Optional
-from uuid import UUID
+from typing import Optional
 
 from nora_lib.interactions.interactions_service import InteractionsService
-from nora_lib.interactions.models import (
-    ReturnedMessage,
-    ReturnedAgentContextMessage,
-    ReturnedAgentContextEvent,
-    EventType,
-    AgentMessageData,
-    Event,
-)
-from nora_lib.context.models import WrappedTaskObject
+from nora_lib.interactions.models import ReturnedMessage
 
 
 class ContextService:

{nora_lib-0.0.7.dev1 → nora_lib-0.1.7}/nora_lib/interactions/interactions_service.py

@@ -1,17 +1,13 @@
-from datetime import datetime
-import logging
 import requests
-from typing import List, Optional
+from typing import Optional
 import json
 
 from nora_lib.interactions.models import (
     AnnotationBatch,
     Event,
-    EventType,
     Message,
     ReturnedMessage,
     ThreadRelationsResponse,
-    ThreadForkEventData,
 )
 
 
@@ -83,11 +79,18 @@ class InteractionsService:
 
         return res
 
-    def fetch_all_threads_by_channel(self, channel_id: str, min_timestamp: str) -> dict:
+    def fetch_all_threads_by_channel(
+        self,
+        channel_id: str,
+        min_timestamp: str,
+        thread_event_types: Optional[list[str]] = None,
+    ) -> dict:
         """Fetch a message from the Interactions API"""
         message_url = f"{self.base_url}/interaction/v1/search/channel"
         request_body = self._channel_lookup_request(
-            channel_id=channel_id, min_timestamp=min_timestamp
+            channel_id=channel_id,
+            min_timestamp=min_timestamp,
+            thread_event_types=thread_event_types,
         )
         response = requests.post(
             message_url,
@@ -98,42 +101,6 @@ class InteractionsService:
         response.raise_for_status()
         return response.json()
 
-    def fetch_messages_and_agent_context_events_for_thread(
-        self, message_id: str, event_type: str
-    ) -> List[ReturnedMessage]:
-        """Build a history of messages for a given message including associated events.
-        This includes messages from pre-forked threads."""
-        messages_with_events: List[ReturnedMessage] = []
-
-        messages_for_thread: ThreadRelationsResponse = (
-            self.fetch_thread_messages_and_events_for_message(
-                message_id, [event_type, EventType.THREAD_FORK.value]
-            )
-        )
-        messages_with_events.extend(messages_for_thread.messages)
-
-        # Process any thread_fork events
-        try:
-            for msg in messages_for_thread.messages:
-                for event in msg.events:
-                    if event.type == EventType.THREAD_FORK.value:
-                        event_data = ThreadForkEventData.model_validate(event.data)
-                        forked_thread: ThreadRelationsResponse = (
-                            self.fetch_thread_messages_and_events_for_message(
-                                event_data.previous_message_id, [event_type]
-                            )
-                        )
-                        messages_with_events.extend(forked_thread.messages)
-        except Exception as e:  # pylint: disable=broad-except
-            logging.exception(
-                "Failed to fetch forked thread messages for message %s: %s",
-                message_id,
-                e,
-            )
-
-        messages_with_events.sort(key=lambda x: x.ts)
-        return messages_with_events
-
     def fetch_thread_messages_and_events_for_message(
         self, message_id: str, event_types: list[str]
     ) -> ThreadRelationsResponse:
@@ -182,41 +149,6 @@ class InteractionsService:
         response.raise_for_status()
         return response.json()
 
-    def fetch_messages_and_events_for_forked_thread(
-        self, message_id: str, event_type: str
-    ) -> List[ReturnedMessage]:
-        """Build a history of messages for a given message including associated events.
-        This includes messages from pre-forked threads."""
-        returned_messages: List[ReturnedMessage] = []
-
-        messages_for_thread: ThreadRelationsResponse = (
-            self.fetch_thread_messages_and_events_for_message(message_id, [event_type])
-        )
-        if messages_for_thread.messages:
-            returned_messages.extend(messages_for_thread.messages)
-
-        # Lookup any thread_fork events (conversation across surfaces)
-        thread_fork_events = self.fetch_messages_and_events_for_thread(
-            messages_for_thread.thread_id, EventType.THREAD_FORK.value
-        )
-        for forked_thread_event in thread_fork_events.get("thread", {}).get(
-            "events", []
-        ):
-            event_data = ThreadForkEventData.model_validate(
-                forked_thread_event.get("data", {})
-            )
-            forked_thread: ThreadRelationsResponse = (
-                self.fetch_thread_messages_and_events_for_message(
-                    event_data.previous_message_id, [event_type]
-                )
-            )
-            if forked_thread.messages:
-                returned_messages.extend(forked_thread.messages)
-
-        returned_messages.sort(key=lambda x: x.ts)
-
-        return returned_messages
-
     def fetch_events_for_message(
         self,
         message_id: str,
@@ -241,7 +173,9 @@ class InteractionsService:
         return response.json()
 
     @staticmethod
-    def _channel_lookup_request(channel_id: str, min_timestamp: str) -> dict:
+    def _channel_lookup_request(
+        channel_id: str, min_timestamp: str, thread_event_types: Optional[list[str]]
+    ) -> dict:
         """Interaction service API request to get threads and messages for a channel"""
         return {
             "id": channel_id,
@@ -251,7 +185,8 @@ class InteractionsService:
                         "messages": {
                             "filter": {"min_timestamp": min_timestamp},
                             "apply_annotations_from_actors": ["*"],
-                        }
+                        },
+                        "events": {"filter": {"type": thread_event_types or []}},
                     }
                 }
             },

{nora_lib-0.0.7.dev1 → nora_lib-0.1.7}/nora_lib/interactions/models.py

@@ -55,15 +55,6 @@ class Message(BaseModel):
         return ts.isoformat()
 
 
-class EventType(str, Enum):
-    """Event types for the interactions service"""
-
-    AGENT_CONTEXT = "agent:message_context"
-    S2_ANNOTATION = "s2_annotation"
-    THREAD_FORK = "thread_fork"
-    UI_INTERACTION = "ui_interaction"
-
-
 class Event(BaseModel):
     """event object to be sent to the interactions service; requires association with a message, thread or channel id"""
 
@@ -141,12 +132,6 @@ class ReturnedAgentContextMessage(BaseModel):
     events: List[ReturnedAgentContextEvent] = Field(default_factory=list)
 
 
-class ThreadForkEventData(BaseModel):
-    """Event data for a thread fork event"""
-
-    previous_message_id: str
-
-
 class ThreadRelationsResponse(BaseModel):
     """Thread format returned by interaction service for thread relations in a search response"""
 

{nora_lib-0.0.7.dev1 → nora_lib-0.1.7}/nora_lib.egg-info/PKG-INFO

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

{nora_lib-0.0.7.dev1 → nora_lib-0.1.7}/setup.py

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