proscenium 0.0.3__tar.gz → 0.0.5__tar.gz

{proscenium-0.0.3 → proscenium-0.0.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: proscenium
-Version: 0.0.3
+Version: 0.0.5
 Summary: Frame AI Agents
 License: ASFv2
 Author: Adam Pingel
@@ -15,7 +15,6 @@ Provides-Extra: testing
 Requires-Dist: aisuite[openai,anthropic] (>=0.1.10,<0.2.0)
 Requires-Dist: datasets (>=3.3.2,<4.0.0)
 Requires-Dist: docstring_parser (>=0.16,<0.17)
-Requires-Dist: eyecite (>=2.6.11,<3.0.0)
 Requires-Dist: gofannon (>=0.25.13,<0.26.0)
 Requires-Dist: langchain-community (>=0.3.18,<0.4.0)
 Requires-Dist: langchain-huggingface (>=0.1.2,<0.2.0)

{proscenium-0.0.3 → proscenium-0.0.5}/proscenium/admin/__init__.py

@@ -28,6 +28,9 @@ class Admin(Character):
         super().__init__(admin_channel_id)
         self.admin_channel_id = admin_channel_id
 
+    def wants_to_handle(self, channel_id: str, speaker_id: str, utterance: str) -> bool:
+        return False
+
     def handle(
         channel_id: str,
         speaker_id: str,

{proscenium-0.0.3 → proscenium-0.0.5}/proscenium/core/__init__.py

@@ -1,12 +1,28 @@
 from typing import Generator
 from typing import Optional
 import logging
+
+from pydantic import BaseModel, Field
 from rich.console import Console
 
 logging.getLogger(__name__).addHandler(logging.NullHandler())
 
 log = logging.getLogger(__name__)
 
+control_flow_system_prompt = """
+You control the workflow of an AI assistant.  You evaluate user-posted messages and decide what the next step is.
+"""
+
+
+class WantsToHandleResponse(BaseModel):
+    """
+    The response to whether the Character wants to handle the provided utterance.
+    """
+
+    wants_to_handle: bool = Field(
+        description="A boolean indicating whether the Character wants to handle the provided utterance.",
+    )
+
 
 class Prop:
     """
@@ -52,6 +68,9 @@ class Character:
     def curtain_up_message(self) -> str:
         return f"- {self.name()}, {self.description().strip()}"
 
+    def wants_to_handle(self, channel_id: str, speaker_id: str, utterance: str) -> bool:
+        return False
+
     def handle(
         channel_id: str, speaker_id: str, utterance: str
     ) -> Generator[tuple[str, str], None, None]:

{proscenium-0.0.3 → proscenium-0.0.5}/proscenium/interfaces/slack.py

@@ -93,31 +93,45 @@ def make_slack_listener(
 
                     # TODO determine whether the handler has a good chance of being useful
 
-                    for receiving_channel_id, response in character.handle(
-                        channel_id, speaker_id, text
-                    ):
-                        response_response = client.web_client.chat_postMessage(
-                            channel=receiving_channel_id, text=response
-                        )
+                    if character.wants_to_handle(channel_id, speaker_id, text):
+
                         log.info(
-                            "Response sent to channel %s",
-                            receiving_channel_id,
+                            "Handler %s in channel %s wants to handle it",
+                            character.name(),
+                            channel_id,
                         )
-                        if receiving_channel_id == admin_channel_id:
-                            continue
 
-                        permalink = client.web_client.chat_getPermalink(
-                            channel=receiving_channel_id,
-                            message_ts=response_response["ts"],
-                        )["permalink"]
+                        for receiving_channel_id, response in character.handle(
+                            channel_id, speaker_id, text
+                        ):
+                            response_response = client.web_client.chat_postMessage(
+                                channel=receiving_channel_id, text=response
+                            )
+                            log.info(
+                                "Response sent to channel %s",
+                                receiving_channel_id,
+                            )
+                            if receiving_channel_id == admin_channel_id:
+                                continue
+
+                            permalink = client.web_client.chat_getPermalink(
+                                channel=receiving_channel_id,
+                                message_ts=response_response["ts"],
+                            )["permalink"]
+                            log.info(
+                                "Response sent to channel %s link %s",
+                                receiving_channel_id,
+                                permalink,
+                            )
+                            client.web_client.chat_postMessage(
+                                channel=admin_channel_id,
+                                text=permalink,
+                            )
+                    else:
                         log.info(
-                            "Response sent to channel %s link %s",
-                            receiving_channel_id,
-                            permalink,
-                        )
-                        client.web_client.chat_postMessage(
-                            channel=admin_channel_id,
-                            text=permalink,
+                            "Handler %s in channel %s does not want to handle it",
+                            character.name(),
+                            channel_id,
                         )
 
         elif req.type == "interactive":
@@ -178,8 +192,8 @@ def bot_user_id(socket_mode_client: SocketModeClient, console: Console):
 
     console.print(auth_response["url"])
     console.print()
-    console.print(f"Team '{auth_response["team"]}' ({auth_response["team_id"]})")
-    console.print(f"User '{auth_response["user"]}' ({auth_response["user_id"]})")
+    console.print("Team", auth_response["team"], auth_response["team_id"])
+    console.print("User", auth_response["user"], auth_response["user_id"])
 
     user_id = auth_response["user_id"]
     console.print("Bot id", auth_response["bot_id"])

{proscenium-0.0.3 → proscenium-0.0.5}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "proscenium"
-version = "0.0.3"
+version = "0.0.5"
 description = "Frame AI Agents"
 authors = ["Adam Pingel <oss@pingel.org>"]
 license = "ASFv2"
@@ -24,7 +24,6 @@ neo4j = "^5.28.1"
 gofannon = "^0.25.13"
 langchain-community = "^0.3.18"
 langchain-huggingface = "^0.1.2"
-eyecite = "^2.6.11"
 slack_sdk = "^3.35.0"
 
 [tool.poetry.extras]