proscenium 0.0.13__py3-none-any.whl → 0.0.14__py3-none-any.whl

proscenium/__init__.py

@@ -1,3 +1,182 @@
+from typing import Generator
+from typing import Optional
 import logging
 
+from pydantic import BaseModel, Field
+from rich.text import Text
+from rich.console import Console
+
 logging.getLogger(__name__).addHandler(logging.NullHandler())
+
+log = logging.getLogger(__name__)
+
+
+def header() -> Text:
+    text = Text()
+    text.append("Proscenium 🎭\n", style="bold")
+    text.append("https://the-ai-alliance.github.io/proscenium/\n")
+    # TODO version, timestamp, ...
+    return text
+
+
+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:
+    """
+    A `Prop` is a resource available to the `Character`s in a `Scene`.
+    """
+
+    def __init__(
+        self,
+        console: Optional[Console] = None,
+    ):
+        self.console = console
+
+    def name(self) -> str:
+        return self.__class__.__name__
+
+    def description(self) -> str:
+        return self.__doc__ or ""
+
+    def curtain_up_message(self) -> str:
+        return f"- {self.name()}, {self.description().strip()}"
+
+    def already_built(self) -> bool:
+        return False
+
+    def build(self) -> None:
+        pass
+
+
+class Character:
+    """
+    A `Character` is a participant in a `Scene` that `handle`s utterances from the
+    scene by producing its own utterances."""
+
+    def __init__(self, admin_channel_id: str):
+        self.admin_channel_id = admin_channel_id
+
+    def name(self) -> str:
+        return self.__class__.__name__
+
+    def description(self) -> str:
+        return self.__doc__ or ""
+
+    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]:
+        pass
+
+
+class Scene:
+    """
+    A `Scene` is a setting in which `Character`s interact with each other and
+    with `Prop`s. It is a container for `Character`s and `Prop`s.
+    """
+
+    def __init__(self):
+        pass
+
+    def name(self) -> str:
+        return self.__class__.__name__
+
+    def description(self) -> str:
+        return self.__doc__ or ""
+
+    def curtain_up_message(self) -> str:
+
+        characters_msg = "\n".join(
+            [character.curtain_up_message() for character in self.characters()]
+        )
+
+        props_msg = "\n".join([prop.curtain_up_message() for prop in self.props()])
+
+        return f"""
+Scene: {self.name()}, {self.description().strip()}
+
+Characters:
+{characters_msg}
+
+Props:
+{props_msg}
+"""
+
+    def props(self) -> list[Prop]:
+        return []
+
+    def prepare_props(self, force_rebuild: bool = False) -> None:
+        for prop in self.props():
+            if force_rebuild:
+                prop.build()
+            elif not prop.already_built():
+                log.info("Prop %s not built. Building it now.", prop.name())
+                prop.build()
+
+    def characters(self) -> list[Character]:
+        return []
+
+    def places(self) -> dict[str, Character]:
+        pass
+
+    def curtain(self) -> None:
+        pass
+
+
+class Production:
+    """
+    A `Production` is a collection of `Scene`s."""
+
+    def __init__(self, admin_channel_id: str, console: Optional[Console] = None):
+        self.admin_channel_id = admin_channel_id
+        self.console = console
+
+    def name(self) -> str:
+        return self.__class__.__name__
+
+    def description(self) -> str:
+        return self.__doc__ or ""
+
+    def prepare_props(self, force_rebuild: bool = False) -> None:
+        if force_rebuild:
+            log.info("Forcing rebuild of all props.")
+        else:
+            log.info("Building any missing props...")
+
+        for scene in self.scenes():
+            scene.prepare_props(force_rebuild=force_rebuild)
+
+    def curtain_up_message(self) -> str:
+
+        scenes_msg = "\n\n".join(
+            [scene.curtain_up_message() for scene in self.scenes()]
+        )
+
+        return f"""Production: {self.name()}, {self.description().strip()}
+
+{scenes_msg}"""
+
+    def scenes(self) -> list[Scene]:
+        return []
+
+    def curtain(self) -> None:
+        for scene in self.scenes():
+            scene.curtain()

proscenium/admin/__init__.py

@@ -4,8 +4,8 @@ from typing import Optional
 
 import logging
 
-from proscenium.core import Prop
-from proscenium.core import Character
+from proscenium import Prop
+from proscenium import Character
 from rich.console import Console
 
 logging.getLogger(__name__).addHandler(logging.NullHandler())

proscenium/bin/__init__.py

@@ -5,7 +5,7 @@ import importlib
 import yaml
 from pathlib import Path
 from rich.console import Console
-from proscenium.core import Production
+from proscenium import Production
 
 logging.getLogger(__name__).addHandler(logging.NullHandler())
 

proscenium/bin/bot.py

@@ -8,7 +8,7 @@ import logging
 from pathlib import Path
 from rich.console import Console
 
-from proscenium.verbs.display import header
+from proscenium import header
 from proscenium.bin import production_from_config
 from proscenium.interfaces.slack import SlackProductionProcessor
 

proscenium/complete.py

@@ -0,0 +1,97 @@
+"""
+This module uses the [`aisuite`](https://github.com/andrewyng/aisuite) library
+to interact with various LLM inference providers.
+
+It provides functions to complete a simple chat prompt, evaluate a tool call,
+and apply a list of tool calls to a chat prompt.
+
+Providers tested with Proscenium include:
+
+# AWS
+
+Environment: `AWS_ACCESS_KEY_ID`, `AWS_SECRET_ACCESS_KEY`
+
+Valid model ids:
+- `aws:meta.llama3-1-8b-instruct-v1:0`
+
+# Anthropic
+
+Environment: `ANTHROPIC_API_KEY`
+
+Valid model ids:
+- `anthropic:claude-3-5-sonnet-20240620`
+
+# OpenAI
+
+Environment: `OPENAI_API_KEY`
+
+Valid model ids:
+- `openai:gpt-4o`
+
+# Ollama
+
+Command line, eg `ollama run llama3.2 --keepalive 2h`
+
+Valid model ids:
+- `ollama:llama3.2`
+- `ollama:granite3.1-dense:2b`
+"""
+
+import logging
+
+from rich.console import Group
+from rich.panel import Panel
+from rich.table import Table
+from rich.text import Text
+
+from aisuite import Client as AISuiteClient
+
+log = logging.getLogger(__name__)
+
+
+def complete_simple(
+    chat_completion_client: AISuiteClient,
+    model_id: str,
+    system_prompt: str,
+    user_prompt: str,
+    **kwargs,
+) -> str:
+
+    console = kwargs.pop("console", None)
+
+    messages = [
+        {"role": "system", "content": system_prompt},
+        {"role": "user", "content": user_prompt},
+    ]
+
+    if console is not None:
+
+        kwargs_text = "\n".join([str(k) + ": " + str(v) for k, v in kwargs.items()])
+
+        params_text = Text(
+            f"""
+model_id: {model_id}
+{kwargs_text}
+    """
+        )
+
+        messages_table = Table(title="Messages", show_lines=True)
+        messages_table.add_column("Role", justify="left")
+        messages_table.add_column("Content", justify="left")  # style="green"
+        for message in messages:
+            messages_table.add_row(message["role"], message["content"])
+
+        call_panel = Panel(
+            Group(params_text, messages_table), title="complete_simple call"
+        )
+        console.print(call_panel)
+
+    response = chat_completion_client.chat.completions.create(
+        model=model_id, messages=messages, **kwargs
+    )
+    response = response.choices[0].message.content
+
+    if console is not None:
+        console.print(Panel(response, title="Response"))
+
+    return response

proscenium/{verbs/display/chat.py → history.py}

@@ -1,5 +1,19 @@
+import logging
+
 from rich.table import Table
 
+log = logging.getLogger(__name__)
+
+
+def format_chat_history(chat_history) -> str:
+    delimiter = "-" * 80 + "\n"
+    return delimiter.join(
+        [
+            f"{msg['sender']} to {msg['receiver']}:\n\n{msg['content']}\n\n"
+            for msg in chat_history
+        ]
+    )
+
 
 def messages_table(messages: list) -> Table:
 

proscenium/interfaces/slack.py

@@ -12,8 +12,8 @@ from slack_sdk.socket_mode import SocketModeClient
 from slack_sdk.socket_mode.request import SocketModeRequest
 from slack_sdk.socket_mode.response import SocketModeResponse
 
-from proscenium.core import Production
-from proscenium.core import Character
+from proscenium import Production
+from proscenium import Character
 from proscenium.admin import Admin
 
 log = logging.getLogger(__name__)

proscenium/patterns/rag.py

@@ -6,7 +6,7 @@ from rich.table import Table
 from pymilvus import MilvusClient
 from pymilvus import model
 
-from proscenium.verbs.complete import complete_simple
+from proscenium.complete import complete_simple
 
 log = logging.getLogger(__name__)
 

proscenium/patterns/tools.py

@@ -1,19 +1,194 @@
 from typing import Optional
+from typing import Any
 import logging
+import json
 
-from rich.console import Console
-from rich.panel import Panel
+from rich.console import Group
+from rich.table import Table
 from rich.text import Text
+from rich.panel import Panel
+from rich.console import Console
+
+from aisuite import Client as AISuiteClient
+from aisuite.framework.message import ChatCompletionMessageToolCall
 
-from proscenium.verbs.complete import (
-    complete_for_tool_applications,
-    evaluate_tool_calls,
-    complete_with_tool_results,
-)
+from gofannon.base import BaseTool
+
+from proscenium.history import messages_table
 
 log = logging.getLogger(__name__)
 
 
+def evaluate_tool_call(tool_map: dict, tool_call: ChatCompletionMessageToolCall) -> Any:
+
+    function_name = tool_call.function.name
+    # TODO validate the arguments?
+    function_args = json.loads(tool_call.function.arguments)
+
+    log.info(f"Evaluating tool call: {function_name} with args {function_args}")
+
+    function_response = tool_map[function_name](**function_args)
+
+    log.info(f"   Response: {function_response}")
+
+    return function_response
+
+
+def tool_response_message(
+    tool_call: ChatCompletionMessageToolCall, tool_result: Any
+) -> dict:
+
+    return {
+        "role": "tool",
+        "tool_call_id": tool_call.id,
+        "name": tool_call.function.name,
+        "content": json.dumps(tool_result),
+    }
+
+
+def evaluate_tool_calls(tool_call_message, tool_map: dict) -> list[dict]:
+
+    tool_call: ChatCompletionMessageToolCall
+
+    log.info("Evaluating tool calls")
+
+    new_messages: list[dict] = []
+
+    for tool_call in tool_call_message.tool_calls:
+        function_response = evaluate_tool_call(tool_map, tool_call)
+        new_messages.append(tool_response_message(tool_call, function_response))
+
+    log.info("Tool calls evaluated")
+
+    return new_messages
+
+
+def complete_for_tool_applications(
+    chat_completion_client: AISuiteClient,
+    model_id: str,
+    messages: list,
+    tool_desc_list: list,
+    temperature: float,
+    console: Optional[Console] = None,
+):
+
+    if console is not None:
+        panel = complete_with_tools_panel(
+            "complete for tool applications",
+            model_id,
+            tool_desc_list,
+            messages,
+            temperature,
+        )
+        console.print(panel)
+
+    response = chat_completion_client.chat.completions.create(
+        model=model_id,
+        messages=messages,
+        temperature=temperature,
+        tools=tool_desc_list,  # tool_choice="auto",
+    )
+
+    return response
+
+
+def complete_with_tool_results(
+    chat_completion_client: AISuiteClient,
+    model_id: str,
+    messages: list,
+    tool_call_message: dict,
+    tool_evaluation_messages: list[dict],
+    tool_desc_list: list,
+    temperature: float,
+    console: Optional[Console] = None,
+):
+
+    messages.append(tool_call_message)
+    messages.extend(tool_evaluation_messages)
+
+    if console is not None:
+        panel = complete_with_tools_panel(
+            "complete call with tool results",
+            model_id,
+            tool_desc_list,
+            messages,
+            temperature,
+        )
+        console.print(panel)
+
+    response = chat_completion_client.chat.completions.create(
+        model=model_id,
+        messages=messages,
+    )
+
+    return response.choices[0].message.content
+
+
+def process_tools(tools: list[BaseTool]) -> tuple[dict, list]:
+    applied_tools = [F() for F in tools]
+    tool_map = {f.name: f.fn for f in applied_tools}
+    tool_desc_list = [f.definition for f in applied_tools]
+    return tool_map, tool_desc_list
+
+
+def parameters_table(parameters: list[dict]) -> Table:
+
+    table = Table(title="Parameters", show_lines=False, box=None)
+    table.add_column("name", justify="right")
+    table.add_column("type", justify="left")
+    table.add_column("description", justify="left")
+
+    for name, props in parameters["properties"].items():
+        table.add_row(name, props["type"], props["description"])
+
+    # TODO denote required params
+
+    return table
+
+
+def function_description_panel(fd: dict) -> Panel:
+
+    fn = fd["function"]
+
+    text = Text(f"{fd['type']} {fn['name']}: {fn['description']}\n")
+
+    pt = parameters_table(fn["parameters"])
+
+    panel = Panel(Group(text, pt))
+
+    return panel
+
+
+def function_descriptions_panel(function_descriptions: list[dict]) -> Panel:
+
+    sub_panels = [function_description_panel(fd) for fd in function_descriptions]
+
+    panel = Panel(Group(*sub_panels), title="Function Descriptions")
+
+    return panel
+
+
+def complete_with_tools_panel(
+    title: str, model_id: str, tool_desc_list: list, messages: list, temperature: float
+) -> Panel:
+
+    text = Text(
+        f"""
+model_id: {model_id}
+temperature: {temperature}
+"""
+    )
+
+    panel = Panel(
+        Group(
+            text, function_descriptions_panel(tool_desc_list), messages_table(messages)
+        ),
+        title=title,
+    )
+
+    return panel
+
+
 def apply_tools(
     model_id: str,
     system_message: str,

{proscenium-0.0.13.dist-info → proscenium-0.0.14.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: proscenium
-Version: 0.0.13
+Version: 0.0.14
 Summary: Declare Collaborative, Asynchronous Human To Agent Interactions
 Author-email: Adam Pingel <oss@pingel.org>
 License-Expression: Apache-2.0
@@ -25,15 +25,17 @@ Dynamic: license-file
 
 # Proscenium
 
-[![PyPI version](https://img.shields.io/pypi/v/proscenium.svg)](https://pypi.org/project/proscenium/)
+[![PyPI version](https://img.shields.io/pypi/v/proscenium)](https://pypi.org/project/proscenium/)
 [![PyPI - Downloads](https://img.shields.io/pypi/dm/proscenium)](https://pypi.org/project/proscenium/)
 [![CI](https://github.com/The-AI-Alliance/proscenium/actions/workflows/pytest.yml/badge.svg)](https://github.com/The-AI-Alliance/proscenium/actions/workflows/pytest.yml)
-[![PyPI](https://img.shields.io/pypi/v/proscenium)](https://pypi.org/project/proscenium/)
 [![License](https://img.shields.io/github/license/The-AI-Alliance/proscenium)](https://github.com/The-AI-Alliance/proscenium/tree/main?tab=Apache-2.0-1-ov-file#readme)
 [![Issues](https://img.shields.io/github/issues/The-AI-Alliance/proscenium)](https://github.com/The-AI-Alliance/proscenium/issues)
 [![GitHub stars](https://img.shields.io/github/stars/The-AI-Alliance/proscenium?style=social)](https://github.com/The-AI-Alliance/proscenium/stargazers)
 
-Proscenium is a small, experimental library of composable glue that allows for succinct construction of enterprise AI applications. It was started in February 2025 and is still in early development.
+Proscenium is a small, experimental library for
+declaring collaborative, asynchronous human to agent interactions
+in enterprise AI applications.
+It was started in February 2025 and is still in early development.
 
 See the [website](https://the-ai-alliance.github.io/proscenium/) for quickstart info, goals, and other links.
 

proscenium-0.0.14.dist-info/RECORD

@@ -0,0 +1,19 @@
+proscenium/__init__.py,sha256=n6X5CqhlDL3lagVL0756myyNwVxHyGokngxkbRzRkxA,4574
+proscenium/complete.py,sha256=WVp-WB420jzikuUmPYVRQ0wI3bgRGZO_-ZCBBX_yJbw,2239
+proscenium/history.py,sha256=AzLfn-raqlDNEJrKQYUJNN8kMogr08evfjOi9Mzwlqo,1445
+proscenium/admin/__init__.py,sha256=5ENOL6CqlZJilCZFt3O6gwbHSvFMFcxEf9f8e2QmcTM,1016
+proscenium/bin/__init__.py,sha256=RcD50sEH58LMEayUez9uugTigG7XnFB6kdILvro_xzg,1155
+proscenium/bin/bot.py,sha256=IaQP33AOCTBcNtx4gb7kCwa2aXTR8XuKuLAfYwPJtE4,1699
+proscenium/interfaces/__init__.py,sha256=nDWNd6_TSf4vDQuHVBoAf4QfZCB3ZUFQ0M7XvifNJ-g,78
+proscenium/interfaces/slack.py,sha256=ebx3oc2CJuzbCJ9eYyCRyhtbojaCaWsgfscmwVv8I3E,10195
+proscenium/patterns/__init__.py,sha256=nDWNd6_TSf4vDQuHVBoAf4QfZCB3ZUFQ0M7XvifNJ-g,78
+proscenium/patterns/graph_rag.py,sha256=1HH1xdlAA6ypvYdP4dWFm-KXrGPUmm0T4qIdAU8mgvE,1763
+proscenium/patterns/rag.py,sha256=HJzLYIqcxhAZNQ0JXhfHe9JV1rr41m9iFQGSxjryGBg,2304
+proscenium/patterns/tools.py,sha256=CjVSySlWqd308ToYiwVSZG-NKnvUlWIGYfh-GY5BXMk,6164
+proscenium/util/__init__.py,sha256=FC1hjA37VvmVpF9-OlYNp9TjArH6etr6KiAvF9t_6lI,679
+proscenium-0.0.14.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+proscenium-0.0.14.dist-info/METADATA,sha256=6SRKhXXMTp7Zpt0Q5kQ7KtwBHF2pJ8dDdiJFr31yxRE,2148
+proscenium-0.0.14.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+proscenium-0.0.14.dist-info/entry_points.txt,sha256=bbrDWv4WKIx9UROGXcyH0eo40F8SNpDQKvXxoE8BTHI,58
+proscenium-0.0.14.dist-info/top_level.txt,sha256=aiTgQy73e21wnSSxM5iICXmXT3zldN8cFKeXpvJ5Xx4,11
+proscenium-0.0.14.dist-info/RECORD,,

proscenium/core/__init__.py

@@ -1,172 +0,0 @@
-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:
-    """
-    A `Prop` is a resource available to the `Character`s in a `Scene`.
-    """
-
-    def __init__(
-        self,
-        console: Optional[Console] = None,
-    ):
-        self.console = console
-
-    def name(self) -> str:
-        return self.__class__.__name__
-
-    def description(self) -> str:
-        return self.__doc__ or ""
-
-    def curtain_up_message(self) -> str:
-        return f"- {self.name()}, {self.description().strip()}"
-
-    def already_built(self) -> bool:
-        return False
-
-    def build(self) -> None:
-        pass
-
-
-class Character:
-    """
-    A `Character` is a participant in a `Scene` that `handle`s utterances from the
-    scene by producing its own utterances."""
-
-    def __init__(self, admin_channel_id: str):
-        self.admin_channel_id = admin_channel_id
-
-    def name(self) -> str:
-        return self.__class__.__name__
-
-    def description(self) -> str:
-        return self.__doc__ or ""
-
-    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]:
-        pass
-
-
-class Scene:
-    """
-    A `Scene` is a setting in which `Character`s interact with each other and
-    with `Prop`s. It is a container for `Character`s and `Prop`s.
-    """
-
-    def __init__(self):
-        pass
-
-    def name(self) -> str:
-        return self.__class__.__name__
-
-    def description(self) -> str:
-        return self.__doc__ or ""
-
-    def curtain_up_message(self) -> str:
-
-        characters_msg = "\n".join(
-            [character.curtain_up_message() for character in self.characters()]
-        )
-
-        props_msg = "\n".join([prop.curtain_up_message() for prop in self.props()])
-
-        return f"""
-Scene: {self.name()}, {self.description().strip()}
-
-Characters:
-{characters_msg}
-
-Props:
-{props_msg}
-"""
-
-    def props(self) -> list[Prop]:
-        return []
-
-    def prepare_props(self, force_rebuild: bool = False) -> None:
-        for prop in self.props():
-            if force_rebuild:
-                prop.build()
-            elif not prop.already_built():
-                log.info("Prop %s not built. Building it now.", prop.name())
-                prop.build()
-
-    def characters(self) -> list[Character]:
-        return []
-
-    def places(self) -> dict[str, Character]:
-        pass
-
-    def curtain(self) -> None:
-        pass
-
-
-class Production:
-    """
-    A `Production` is a collection of `Scene`s."""
-
-    def __init__(self, admin_channel_id: str, console: Optional[Console] = None):
-        self.admin_channel_id = admin_channel_id
-        self.console = console
-
-    def name(self) -> str:
-        return self.__class__.__name__
-
-    def description(self) -> str:
-        return self.__doc__ or ""
-
-    def prepare_props(self, force_rebuild: bool = False) -> None:
-        if force_rebuild:
-            log.info("Forcing rebuild of all props.")
-        else:
-            log.info("Building any missing props...")
-
-        for scene in self.scenes():
-            scene.prepare_props(force_rebuild=force_rebuild)
-
-    def curtain_up_message(self) -> str:
-
-        scenes_msg = "\n\n".join(
-            [scene.curtain_up_message() for scene in self.scenes()]
-        )
-
-        return f"""Production: {self.name()}, {self.description().strip()}
-
-{scenes_msg}"""
-
-    def scenes(self) -> list[Scene]:
-        return []
-
-    def curtain(self) -> None:
-        for scene in self.scenes():
-            scene.curtain()

proscenium/verbs/__init__.py

@@ -1,3 +0,0 @@
-import logging
-
-logging.getLogger(__name__).addHandler(logging.NullHandler())

proscenium/verbs/complete.py

@@ -1,209 +0,0 @@
-"""
-This module uses the [`aisuite`](https://github.com/andrewyng/aisuite) library
-to interact with various LLM inference providers.
-
-It provides functions to complete a simple chat prompt, evaluate a tool call,
-and apply a list of tool calls to a chat prompt.
-
-Providers tested with Proscenium include:
-
-# AWS
-
-Environment: `AWS_ACCESS_KEY_ID`, `AWS_SECRET_ACCESS_KEY`
-
-Valid model ids:
-- `aws:meta.llama3-1-8b-instruct-v1:0`
-
-# Anthropic
-
-Environment: `ANTHROPIC_API_KEY`
-
-Valid model ids:
-- `anthropic:claude-3-5-sonnet-20240620`
-
-# OpenAI
-
-Environment: `OPENAI_API_KEY`
-
-Valid model ids:
-- `openai:gpt-4o`
-
-# Ollama
-
-Command line, eg `ollama run llama3.2 --keepalive 2h`
-
-Valid model ids:
-- `ollama:llama3.2`
-- `ollama:granite3.1-dense:2b`
-"""
-
-from typing import Optional
-from typing import Any
-import logging
-import json
-
-from rich.console import Console
-from rich.console import Group
-from rich.panel import Panel
-from rich.table import Table
-from rich.text import Text
-
-from aisuite import Client as AISuiteClient
-from aisuite.framework.message import ChatCompletionMessageToolCall
-
-from proscenium.verbs.display.tools import complete_with_tools_panel
-
-log = logging.getLogger(__name__)
-
-
-def complete_simple(
-    chat_completion_client: AISuiteClient,
-    model_id: str,
-    system_prompt: str,
-    user_prompt: str,
-    **kwargs,
-) -> str:
-
-    console = kwargs.pop("console", None)
-
-    messages = [
-        {"role": "system", "content": system_prompt},
-        {"role": "user", "content": user_prompt},
-    ]
-
-    if console is not None:
-
-        kwargs_text = "\n".join([str(k) + ": " + str(v) for k, v in kwargs.items()])
-
-        params_text = Text(
-            f"""
-model_id: {model_id}
-{kwargs_text}
-    """
-        )
-
-        messages_table = Table(title="Messages", show_lines=True)
-        messages_table.add_column("Role", justify="left")
-        messages_table.add_column("Content", justify="left")  # style="green"
-        for message in messages:
-            messages_table.add_row(message["role"], message["content"])
-
-        call_panel = Panel(
-            Group(params_text, messages_table), title="complete_simple call"
-        )
-        console.print(call_panel)
-
-    response = chat_completion_client.chat.completions.create(
-        model=model_id, messages=messages, **kwargs
-    )
-    response = response.choices[0].message.content
-
-    if console is not None:
-        console.print(Panel(response, title="Response"))
-
-    return response
-
-
-def evaluate_tool_call(tool_map: dict, tool_call: ChatCompletionMessageToolCall) -> Any:
-
-    function_name = tool_call.function.name
-    # TODO validate the arguments?
-    function_args = json.loads(tool_call.function.arguments)
-
-    log.info(f"Evaluating tool call: {function_name} with args {function_args}")
-
-    function_response = tool_map[function_name](**function_args)
-
-    log.info(f"   Response: {function_response}")
-
-    return function_response
-
-
-def tool_response_message(
-    tool_call: ChatCompletionMessageToolCall, tool_result: Any
-) -> dict:
-
-    return {
-        "role": "tool",
-        "tool_call_id": tool_call.id,
-        "name": tool_call.function.name,
-        "content": json.dumps(tool_result),
-    }
-
-
-def evaluate_tool_calls(tool_call_message, tool_map: dict) -> list[dict]:
-
-    tool_call: ChatCompletionMessageToolCall
-
-    log.info("Evaluating tool calls")
-
-    new_messages: list[dict] = []
-
-    for tool_call in tool_call_message.tool_calls:
-        function_response = evaluate_tool_call(tool_map, tool_call)
-        new_messages.append(tool_response_message(tool_call, function_response))
-
-    log.info("Tool calls evaluated")
-
-    return new_messages
-
-
-def complete_for_tool_applications(
-    chat_completion_client: AISuiteClient,
-    model_id: str,
-    messages: list,
-    tool_desc_list: list,
-    temperature: float,
-    console: Optional[Console] = None,
-):
-
-    if console is not None:
-        panel = complete_with_tools_panel(
-            "complete for tool applications",
-            model_id,
-            tool_desc_list,
-            messages,
-            temperature,
-        )
-        console.print(panel)
-
-    response = chat_completion_client.chat.completions.create(
-        model=model_id,
-        messages=messages,
-        temperature=temperature,
-        tools=tool_desc_list,  # tool_choice="auto",
-    )
-
-    return response
-
-
-def complete_with_tool_results(
-    chat_completion_client: AISuiteClient,
-    model_id: str,
-    messages: list,
-    tool_call_message: dict,
-    tool_evaluation_messages: list[dict],
-    tool_desc_list: list,
-    temperature: float,
-    console: Optional[Console] = None,
-):
-
-    messages.append(tool_call_message)
-    messages.extend(tool_evaluation_messages)
-
-    if console is not None:
-        panel = complete_with_tools_panel(
-            "complete call with tool results",
-            model_id,
-            tool_desc_list,
-            messages,
-            temperature,
-        )
-        console.print(panel)
-
-    response = chat_completion_client.chat.completions.create(
-        model=model_id,
-        messages=messages,
-    )
-
-    return response.choices[0].message.content

proscenium/verbs/display/__init__.py

@@ -1,9 +0,0 @@
-from rich.text import Text
-
-
-def header() -> Text:
-    text = Text()
-    text.append("Proscenium 🎭\n", style="bold")
-    text.append("https://the-ai-alliance.github.io/proscenium/\n")
-    # TODO version, timestamp, ...
-    return text

proscenium/verbs/display/tools.py

@@ -1,64 +0,0 @@
-from rich.console import Group
-from rich.table import Table
-from rich.text import Text
-from rich.panel import Panel
-
-from .chat import messages_table
-
-
-def parameters_table(parameters: list[dict]) -> Table:
-
-    table = Table(title="Parameters", show_lines=False, box=None)
-    table.add_column("name", justify="right")
-    table.add_column("type", justify="left")
-    table.add_column("description", justify="left")
-
-    for name, props in parameters["properties"].items():
-        table.add_row(name, props["type"], props["description"])
-
-    # TODO denote required params
-
-    return table
-
-
-def function_description_panel(fd: dict) -> Panel:
-
-    fn = fd["function"]
-
-    text = Text(f"{fd['type']} {fn['name']}: {fn['description']}\n")
-
-    pt = parameters_table(fn["parameters"])
-
-    panel = Panel(Group(text, pt))
-
-    return panel
-
-
-def function_descriptions_panel(function_descriptions: list[dict]) -> Panel:
-
-    sub_panels = [function_description_panel(fd) for fd in function_descriptions]
-
-    panel = Panel(Group(*sub_panels), title="Function Descriptions")
-
-    return panel
-
-
-def complete_with_tools_panel(
-    title: str, model_id: str, tool_desc_list: list, messages: list, temperature: float
-) -> Panel:
-
-    text = Text(
-        f"""
-model_id: {model_id}
-temperature: {temperature}
-"""
-    )
-
-    panel = Panel(
-        Group(
-            text, function_descriptions_panel(tool_desc_list), messages_table(messages)
-        ),
-        title=title,
-    )
-
-    return panel

proscenium/verbs/display.py

@@ -1,13 +0,0 @@
-import logging
-from rich.text import Text
-
-log = logging.getLogger(__name__)
-
-
-def header() -> Text:
-    text = Text(
-        """[bold]Proscenium[/bold] :performing_arts:
-[bold]The AI Alliance[/bold]"""
-    )
-    # TODO version, timestamp, ...
-    return text

proscenium/verbs/invoke.py

@@ -1,11 +0,0 @@
-import logging
-from gofannon.base import BaseTool
-
-log = logging.getLogger(__name__)
-
-
-def process_tools(tools: list[BaseTool]) -> tuple[dict, list]:
-    applied_tools = [F() for F in tools]
-    tool_map = {f.name: f.fn for f in applied_tools}
-    tool_desc_list = [f.definition for f in applied_tools]
-    return tool_map, tool_desc_list

proscenium/verbs/remember.py

@@ -1,13 +0,0 @@
-import logging
-
-log = logging.getLogger(__name__)
-
-
-def format_chat_history(chat_history) -> str:
-    delimiter = "-" * 80 + "\n"
-    return delimiter.join(
-        [
-            f"{msg['sender']} to {msg['receiver']}:\n\n{msg['content']}\n\n"
-            for msg in chat_history
-        ]
-    )

proscenium-0.0.13.dist-info/RECORD

@@ -1,26 +0,0 @@
-proscenium/__init__.py,sha256=nDWNd6_TSf4vDQuHVBoAf4QfZCB3ZUFQ0M7XvifNJ-g,78
-proscenium/admin/__init__.py,sha256=31XUp83v7oCEXTeJb-GgnIitfs2EWovr3QJOewZ0zSg,1026
-proscenium/bin/__init__.py,sha256=ThVsDG6BmnZ86gYaZLGMDcNmR6fVwBLJG3k_JawfzOY,1160
-proscenium/bin/bot.py,sha256=cuHj2-ldPXGRf75Ts_J_y4WfDgTB49iN3sNmnoKKFus,1713
-proscenium/core/__init__.py,sha256=aSUqPMn2LjZ0C_l9Tx6yqqlfCzM7oSljZxHosJyjlLU,4335
-proscenium/interfaces/__init__.py,sha256=nDWNd6_TSf4vDQuHVBoAf4QfZCB3ZUFQ0M7XvifNJ-g,78
-proscenium/interfaces/slack.py,sha256=zfj1D4O6ucVdw6HFnBIMGAbZUb5XK17c-57une42C48,10205
-proscenium/patterns/__init__.py,sha256=nDWNd6_TSf4vDQuHVBoAf4QfZCB3ZUFQ0M7XvifNJ-g,78
-proscenium/patterns/graph_rag.py,sha256=1HH1xdlAA6ypvYdP4dWFm-KXrGPUmm0T4qIdAU8mgvE,1763
-proscenium/patterns/rag.py,sha256=iNPL7i8zfhEGIZAI7FjAs5yKgN1iV_E-PcJiQFrm7lE,2310
-proscenium/patterns/tools.py,sha256=f2CD6f7CYiSs0Tm1Ff1sOL5Ti6DqJ5HQvMI7NmIgqNs,1740
-proscenium/util/__init__.py,sha256=FC1hjA37VvmVpF9-OlYNp9TjArH6etr6KiAvF9t_6lI,679
-proscenium/verbs/__init__.py,sha256=nDWNd6_TSf4vDQuHVBoAf4QfZCB3ZUFQ0M7XvifNJ-g,78
-proscenium/verbs/complete.py,sha256=oRvIt6NR9yhzEHpT1BRKAmf05hVXI91v83A-7OSo-zs,5147
-proscenium/verbs/display.py,sha256=hHFmktyJtjYLi4I1-8HUfmsuoMTIxc6JFfczASBsCbI,260
-proscenium/verbs/invoke.py,sha256=-Bk7Pp0EEwRTS0MJUlViZeUNo8wxnDKJj5q78KU4CdM,339
-proscenium/verbs/remember.py,sha256=Hh9BDRAYf7MGeMD4MzU73p6Q28KrSiLWPx4GjTW1amQ,296
-proscenium/verbs/display/__init__.py,sha256=GXuvaMld8tzfJGngHdwVT-YLnuRmW2G0pMdti9Vj53s,238
-proscenium/verbs/display/chat.py,sha256=2THBUdhG3cIIVZOnJ_AMYL4nWXKFG2cuSkX6wkm48yQ,1148
-proscenium/verbs/display/tools.py,sha256=eR5g-r7MGKFZY0qg-ndkW3p0mfbupV0UaAUFqJPfnNM,1491
-proscenium-0.0.13.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-proscenium-0.0.13.dist-info/METADATA,sha256=50Uqw56v2u2hdl8onbJWYfXSqPMgSdK_0Y6djW68OrA,2229
-proscenium-0.0.13.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-proscenium-0.0.13.dist-info/entry_points.txt,sha256=bbrDWv4WKIx9UROGXcyH0eo40F8SNpDQKvXxoE8BTHI,58
-proscenium-0.0.13.dist-info/top_level.txt,sha256=aiTgQy73e21wnSSxM5iICXmXT3zldN8cFKeXpvJ5Xx4,11
-proscenium-0.0.13.dist-info/RECORD,,