proscenium 0.0.1__py3-none-any.whl → 0.0.3__py3-none-any.whl

proscenium/verbs/read.py

@@ -3,13 +3,18 @@ from typing import List
 import os
 import logging
 
-from langchain_core.documents.base import Document
+import httpx
+from pydantic.networks import HttpUrl
+from pathlib import Path
 
+from langchain_core.documents.base import Document
 from langchain_community.document_loaders import TextLoader
 from langchain_community.document_loaders.hugging_face_dataset import (
     HuggingFaceDatasetLoader,
 )
 
+log = logging.getLogger(__name__)
+
 os.environ["TOKENIZERS_PARALLELISM"] = "false"
 logging.getLogger("langchain_text_splitters.base").setLevel(logging.ERROR)
 
@@ -34,20 +39,13 @@ def load_hugging_face_dataset(
     return documents
 
 
-import httpx
-from pydantic.networks import HttpUrl
-from pathlib import Path
-
-
 async def url_to_file(url: HttpUrl, data_file: Path, overwrite: bool = False):
 
     if data_file.exists() and not overwrite:
-        # print(f"File {data_file} exists. Use overwrite=True to replace.")
         return
 
     async with httpx.AsyncClient() as client:
 
-        # print(f"Downloading {url} to {data_file}...")
         response = await client.get(url)
         response.raise_for_status()
 

proscenium/verbs/remember.py

@@ -1,3 +1,8 @@
+import logging
+
+log = logging.getLogger(__name__)
+
+
 def format_chat_history(chat_history) -> str:
     delimiter = "-" * 80 + "\n"
     return delimiter.join(

proscenium/verbs/vector_database.py

@@ -1,13 +1,17 @@
-from typing import Dict, List, Optional
+from typing import Dict, List
 
+import logging
 from pathlib import Path
 from langchain_core.documents.base import Document
+from urllib.parse import urlsplit
 from pymilvus import MilvusClient
 from pymilvus import DataType, FieldSchema, CollectionSchema
 from pymilvus import model
 
 # See https://milvus.io/docs/quickstart.md
 
+log = logging.getLogger(__name__)
+
 
 def embedding_function(
     embedding_model_id: str,
@@ -39,35 +43,28 @@ def schema_chunks(
     return schema
 
 
-from urllib.parse import urlsplit
-
-
 def vector_db(
     uri: str,
-    overwrite: bool = False,
 ) -> MilvusClient:
 
+    log.info("Connecting to vector db %s", uri)
     uri_fields = urlsplit(uri)
     client = None
     if uri_fields[0] == "file":
         file_path = Path(uri_fields[2][1:])
         if file_path.exists():
-            if overwrite:
-                file_path.unlink()
-                print("Deleted existing vector db file", file_path)
-            else:
-                print(
-                    "Using existing",
-                    uri_fields[2],
-                    "file. Use overwrite=True to replace.",
-                )
+            log.info(
+                "Using existing %s file.",
+                uri_fields[2],
+            )
         else:
-            print("Creating new vector db file", file_path)
+            log.info("Creating new vector db file %s", file_path)
 
         client = MilvusClient(uri=str(file_path))
 
     else:
 
+        log.info("Connecting to vector db at non-file uri %s", uri)
         client = MilvusClient(uri=uri)
 
     return client
@@ -77,12 +74,8 @@ def create_collection(
     client: MilvusClient,
     embedding_fn: model.dense.SentenceTransformerEmbeddingFunction,
     collection_name: str,
-    overwrite: bool = True,
 ) -> None:
 
-    if overwrite and client.has_collection(collection_name):
-        client.drop_collection(collection_name)
-
     client.create_collection(
         collection_name=collection_name,
         schema=schema_chunks(embedding_fn),
@@ -100,7 +93,7 @@ def create_collection(
     client.create_index(
         collection_name=collection_name, index_params=index_params, sync=True
     )
-    print("Created collection", collection_name)
+    log.info("Created collection %s", collection_name)
 
 
 def add_chunks_to_vector_db(

proscenium/verbs/write.py

@@ -1,7 +1,10 @@
 from typing import List
 
+import logging
 import csv
 
+log = logging.getLogger(__name__)
+
 
 def triples_to_csv(triples: List[tuple[str, str, str]], filename: str) -> None:
 

{proscenium-0.0.1.dist-info → proscenium-0.0.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: proscenium
-Version: 0.0.1
+Version: 0.0.3
 Summary: Frame AI Agents
 License: ASFv2
 Author: Adam Pingel
@@ -25,8 +25,8 @@ Requires-Dist: pymilvus (>=2.5.4,<3.0.0)
 Requires-Dist: pymilvus_model (>=0.3.1,<0.4.0)
 Requires-Dist: python-dotenv (>=1.0.1,<2.0.0)
 Requires-Dist: rich (>=13.9.4,<14.0.0)
+Requires-Dist: slack_sdk (>=3.35.0,<4.0.0)
 Requires-Dist: stringcase (>=1.2.0,<2.0.0)
-Requires-Dist: thespian (>=4.0.0,<5.0.0)
 Requires-Dist: tiktoken (>=0.9.0,<0.10.0)
 Requires-Dist: typer (>=0.15.2,<0.16.0)
 Description-Content-Type: text/markdown
@@ -34,17 +34,14 @@ Description-Content-Type: text/markdown
 # 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 is in early development.
+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.
 
-It is also set of simple demonstration applications that elucidate aspects of application and library design.
-
-Currently, proscenium development prioritizes support for domains where
-the creation and use of structured data is critical.
+Currently, proscenium development prioritizes support for domains where the creation and use of structured data is critical.
 
 See the [website](https://the-ai-alliance.github.io/proscenium/) for quickstart info, goals, and other links.
 

proscenium-0.0.3.dist-info/RECORD

@@ -0,0 +1,34 @@
+proscenium/__init__.py,sha256=nDWNd6_TSf4vDQuHVBoAf4QfZCB3ZUFQ0M7XvifNJ-g,78
+proscenium/admin/__init__.py,sha256=VFcFIduWHmcM3idIbWOFGrdxp6oc5ta_ywWIL-hG4Y4,813
+proscenium/bin/bot.py,sha256=nK4WN8ggpCr_KDpDI16Ib8RljmMpJyZ8qZTW8j7lwP4,3700
+proscenium/core/__init__.py,sha256=MFOqGAKIK-XjbKaiFLihAX-zrsoRz4uNfCMahC4nJyc,3620
+proscenium/interfaces/__init__.py,sha256=nDWNd6_TSf4vDQuHVBoAf4QfZCB3ZUFQ0M7XvifNJ-g,78
+proscenium/interfaces/slack.py,sha256=kq4jVsTJUUDIrlO54ZIGqTvJoScZS3dcs1fBoNpg-m4,7947
+proscenium/patterns/__init__.py,sha256=nDWNd6_TSf4vDQuHVBoAf4QfZCB3ZUFQ0M7XvifNJ-g,78
+proscenium/patterns/chunk_space.py,sha256=kQzGUtkQKGQGzGjBqS7jz_zr3uvJeiCBD2leflOenM8,1635
+proscenium/patterns/document_enricher.py,sha256=u-Q4FwvNAFj9nday235jUwxIwizht_VAW8DsmNQmoJs,2374
+proscenium/patterns/entity_resolver.py,sha256=SOK9WO6WFuhSWPFBy4Hj09E2qgqrzxE0MT65nAgZDUw,2748
+proscenium/patterns/graph_rag.py,sha256=1HH1xdlAA6ypvYdP4dWFm-KXrGPUmm0T4qIdAU8mgvE,1763
+proscenium/patterns/knowledge_graph.py,sha256=VLjG8Rp7YfJLZKe9bZt2d4NsGloBV1AYI6SuaQtRLhs,1137
+proscenium/patterns/rag.py,sha256=zvl_P48F3PDfVMgRXeiClLlevMsPKCMA1teVq9X20OE,1494
+proscenium/patterns/tools.py,sha256=f2CD6f7CYiSs0Tm1Ff1sOL5Ti6DqJ5HQvMI7NmIgqNs,1740
+proscenium/verbs/__init__.py,sha256=nDWNd6_TSf4vDQuHVBoAf4QfZCB3ZUFQ0M7XvifNJ-g,78
+proscenium/verbs/chunk.py,sha256=hlVHfuR7sEAR8euh3FRd8hb2eJozE7bHe-E0RmAoFP8,1106
+proscenium/verbs/complete.py,sha256=Y1T49OfAV7K8p0DMzE4aVqtkgVfjUqb6IeOazzdYGow,5071
+proscenium/verbs/display/__init__.py,sha256=GXuvaMld8tzfJGngHdwVT-YLnuRmW2G0pMdti9Vj53s,238
+proscenium/verbs/display/chat.py,sha256=2THBUdhG3cIIVZOnJ_AMYL4nWXKFG2cuSkX6wkm48yQ,1148
+proscenium/verbs/display/milvus.py,sha256=GZze02_ZllukS7zVg0sCPxtjQ4z7O62nFY0Be4d5BX0,2390
+proscenium/verbs/display/neo4j.py,sha256=yT3hulI1U7s4VmHh_UHHHxN25Q_rRgu6KM2so_pTFUI,727
+proscenium/verbs/display/tools.py,sha256=eR5g-r7MGKFZY0qg-ndkW3p0mfbupV0UaAUFqJPfnNM,1491
+proscenium/verbs/display.py,sha256=hHFmktyJtjYLi4I1-8HUfmsuoMTIxc6JFfczASBsCbI,260
+proscenium/verbs/extract.py,sha256=2JSAblCww5Q2wiFXMib-D1iHKm6a5yj7AUYoyGoLQB8,1615
+proscenium/verbs/invoke.py,sha256=-Bk7Pp0EEwRTS0MJUlViZeUNo8wxnDKJj5q78KU4CdM,339
+proscenium/verbs/read.py,sha256=twFtcuyP-y-UwksLmGMCOjMqI7mp--VgvkGDfga6IxA,1262
+proscenium/verbs/remember.py,sha256=Hh9BDRAYf7MGeMD4MzU73p6Q28KrSiLWPx4GjTW1amQ,296
+proscenium/verbs/vector_database.py,sha256=U09P7jnpzUDeP7pEgJubf8xQsxC-O8Qb0MS0KY8eoe8,3527
+proscenium/verbs/write.py,sha256=0GUJuixLnuu_EbFFzAIgrhLEQnOrL0TdUlMiqOl9KtA,367
+proscenium-0.0.3.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+proscenium-0.0.3.dist-info/METADATA,sha256=-CZihFinSAwy-OZz9iGBLKkQRsA3TlyzWy3VYJARvGs,2528
+proscenium-0.0.3.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+proscenium-0.0.3.dist-info/entry_points.txt,sha256=Q05DVkPq_SjgD8mFN6bG5ae2r_UbsqKCdy2kDAtHYGU,57
+proscenium-0.0.3.dist-info/RECORD,,

{proscenium-0.0.1.dist-info → proscenium-0.0.3.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: poetry-core 2.1.2
+Generator: poetry-core 2.1.3
 Root-Is-Purelib: true
 Tag: py3-none-any

proscenium-0.0.3.dist-info/entry_points.txt

@@ -0,0 +1,3 @@
+[console_scripts]
+proscenium-bot=proscenium.bin.bot:app
+

proscenium/scripts/chunk_space.py

@@ -1,33 +0,0 @@
-from rich import print
-
-from pymilvus import MilvusClient
-from pymilvus import model
-
-from proscenium.verbs.read import load_file
-from proscenium.verbs.chunk import documents_to_chunks_by_characters
-from proscenium.verbs.vector_database import create_collection
-from proscenium.verbs.vector_database import add_chunks_to_vector_db
-from proscenium.verbs.display.milvus import collection_panel
-
-
-def build_vector_db(
-    data_files: list[str],
-    vector_db_client: MilvusClient,
-    embedding_fn: model.dense.SentenceTransformerEmbeddingFunction,
-    collection_name: str,
-):
-
-    create_collection(vector_db_client, embedding_fn, collection_name, overwrite=True)
-
-    for data_file in data_files:
-
-        documents = load_file(data_file)
-        chunks = documents_to_chunks_by_characters(documents)
-        print("Data file", data_file, "has", len(chunks), "chunks")
-
-        info = add_chunks_to_vector_db(
-            vector_db_client, embedding_fn, chunks, collection_name
-        )
-        print(info["insert_count"], "chunks inserted")
-
-    print(collection_panel(vector_db_client, collection_name))

proscenium/scripts/graph_rag.py

@@ -1,43 +0,0 @@
-from typing import Callable
-
-from pydantic import BaseModel
-
-from rich import print
-from rich.panel import Panel
-
-from neo4j import Driver
-
-from proscenium.verbs.complete import complete_simple
-
-
-def query_to_prompts(
-    question: str,
-    query_extraction_model_id: str,
-    milvus_uri: str,
-    driver: Driver,
-    query_extract: Callable[
-        [str, str, bool], BaseModel
-    ],  # (query_str, query_extraction_model_id) -> QueryExtractions
-    extract_to_context: Callable[
-        [BaseModel, str, Driver, str, bool], BaseModel
-    ],  # (QueryExtractions, query_str, Driver, milvus_uri) -> Context
-    context_to_prompts: Callable[
-        [BaseModel, bool], tuple[str, str]
-    ],  # Context -> (system_prompt, user_prompt)
-    verbose: bool = False,
-) -> str:
-
-    print("Extracting information from the question")
-    extract = query_extract(question, query_extraction_model_id, verbose)
-    if extract is None:
-        print("Unable to extract information from that question")
-        return None
-    print("Extract:", extract)
-
-    print("Forming context from the extracted information")
-    context = extract_to_context(extract, question, driver, milvus_uri, verbose)
-    print("Context:", context)
-
-    prompts = context_to_prompts(context, verbose)
-
-    return prompts

proscenium/verbs/know.py

@@ -1,9 +0,0 @@
-from neo4j import GraphDatabase
-from neo4j import Driver
-
-
-def knowledge_graph_client(uri: str, username: str, password: str) -> Driver:
-
-    driver = GraphDatabase.driver(uri, auth=(username, password))
-
-    return driver

proscenium-0.0.1.dist-info/RECORD

@@ -1,30 +0,0 @@
-proscenium/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-proscenium/scripts/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-proscenium/scripts/chunk_space.py,sha256=04oXMUe5u-Mes8w6HFhUNkBy5hoXP-p1wHEw81XXOg4,1107
-proscenium/scripts/document_enricher.py,sha256=JkK--QTlmVg7YA8scTm5YItWVWts5YscmpHEU1OeqjI,2214
-proscenium/scripts/entity_resolver.py,sha256=S5-df41ZXaXDQLZ9vfFPJxq6v6fKpBHWZXsDrbD7Gis,2689
-proscenium/scripts/graph_rag.py,sha256=D6lbOb0ftgf5VcUxs9cH6EjlKPBR1zarexNDgI4k4Bw,1287
-proscenium/scripts/knowledge_graph.py,sha256=MddMOW8zkt0wptKijkKUbURpMYz3dAxY1BQklVAgZ1A,1104
-proscenium/scripts/rag.py,sha256=O4qOWGsusqGfb4TMVk3fTav3BmPpToKSGm_2iTxzH2s,1625
-proscenium/scripts/tools.py,sha256=Y4Oq_7vdgppQuDgx_513Wfz-pGDC1ZW5-m8wklmNvD4,2495
-proscenium/verbs/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-proscenium/verbs/chunk.py,sha256=-unQS0xgTyTDsikbueNEvPXQhwK26bQYMZBAeAEvVds,1071
-proscenium/verbs/complete.py,sha256=QcDd0qVISk2NHWyGL6KTH8rzwSSF9SE4UB3Yzp0zrOI,5075
-proscenium/verbs/display/__init__.py,sha256=VEvWaRpHnpE4sLKTdbFe8PybnREn4M4O9XYKRyAsdxk,222
-proscenium/verbs/display/chat.py,sha256=2THBUdhG3cIIVZOnJ_AMYL4nWXKFG2cuSkX6wkm48yQ,1148
-proscenium/verbs/display/huggingface.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-proscenium/verbs/display/milvus.py,sha256=GZze02_ZllukS7zVg0sCPxtjQ4z7O62nFY0Be4d5BX0,2390
-proscenium/verbs/display/neo4j.py,sha256=yT3hulI1U7s4VmHh_UHHHxN25Q_rRgu6KM2so_pTFUI,727
-proscenium/verbs/display/tools.py,sha256=eR5g-r7MGKFZY0qg-ndkW3p0mfbupV0UaAUFqJPfnNM,1491
-proscenium/verbs/display.py,sha256=rCgiLok2EmiHGK8yUBLqxZR148BwroPMqsDf7qQwO1g,210
-proscenium/verbs/extract.py,sha256=YQm7EGjBIP5klVAiG0ZXMXouxf20uUdxShkhGL-Cmw8,1519
-proscenium/verbs/invoke.py,sha256=5szyVIuS3rd2zfyzbWzxVRFr2snf8XDJgzQuhQ85X9Y,289
-proscenium/verbs/know.py,sha256=GF0uvZFn2xbyigdmPoJa1y_NeXspiBhpYWnE7VsG5W8,223
-proscenium/verbs/read.py,sha256=iGTwY0h3cW1_ngyA-JT6HNsVz0BHpEtxamSxDhk97Y8,1361
-proscenium/verbs/remember.py,sha256=AVTVD2Xm0DKAqlhiqHpDf8VFZsoAAgTLqEipZGyqdjo,244
-proscenium/verbs/vector_database.py,sha256=rIxWE_1HZOXrbyb5ICsJFWZZoPax5TJefJ96mSnp6NM,3739
-proscenium/verbs/write.py,sha256=MboUV1828XHITMurtddkm6kXkEgvtzfpBmSLh5X_msc,317
-proscenium-0.0.1.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-proscenium-0.0.1.dist-info/METADATA,sha256=O_1ftoYXXbD68gNsTYgfYRBfhHvXRQ6fIu8v-7QV1dA,2509
-proscenium-0.0.1.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
-proscenium-0.0.1.dist-info/RECORD,,