proscenium 0.0.9__py3-none-any.whl → 0.0.11__py3-none-any.whl

proscenium/patterns/rag.py

@@ -1,12 +1,11 @@
 from typing import List, Dict
+
 import logging
+from rich.table import Table
 
 from pymilvus import MilvusClient
 from pymilvus import model
 
-from lapidarist.verbs.display.milvus import chunk_hits_table
-from lapidarist.verbs.vector_database import closest_chunks
-
 from proscenium.verbs.complete import complete_simple
 
 log = logging.getLogger(__name__)
@@ -38,6 +37,41 @@ def rag_prompt(chunks: List[Dict], query: str) -> str:
     return rag_prompt_template.format(context=context, query=query)
 
 
+def closest_chunks(
+    client: MilvusClient,
+    embedding_fn: model.dense.SentenceTransformerEmbeddingFunction,
+    query: str,
+    collection_name: str,
+    k: int = 4,
+) -> List[Dict]:
+
+    client.load_collection(collection_name)
+
+    result = client.search(
+        collection_name=collection_name,
+        data=embedding_fn.encode_queries([query]),
+        anns_field="vector",
+        search_params={"metric": "IP", "offset": 0},
+        output_fields=["text"],
+        limit=k,
+    )
+
+    hits = result[0]
+
+    return hits
+
+
+def chunk_hits_table(chunks: list[dict]) -> Table:
+
+    table = Table(title="Closest Chunks", show_lines=True)
+    table.add_column("id", justify="right")
+    table.add_column("distance")
+    table.add_column("entity.text", justify="right")
+    for chunk in chunks:
+        table.add_row(str(chunk["id"]), str(chunk["distance"]), chunk["entity"]["text"])
+    return table
+
+
 def answer_question(
     query: str,
     model_id: str,

{proscenium-0.0.9.dist-info → proscenium-0.0.11.dist-info}/METADATA

@@ -1,35 +1,27 @@
-Metadata-Version: 2.3
+Metadata-Version: 2.4
 Name: proscenium
-Version: 0.0.9
+Version: 0.0.11
 Summary: Frame AI Agents
-License: ASFv2
-Author: Adam Pingel
-Author-email: oss@pingel.org
-Requires-Python: >=3.11,<4.0
-Classifier: License :: Other/Proprietary License
-Classifier: Programming Language :: Python :: 3
-Classifier: Programming Language :: Python :: 3.11
-Classifier: Programming Language :: Python :: 3.12
-Classifier: Programming Language :: Python :: 3.13
-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: 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)
-Requires-Dist: lapidarist (>=0.0.1,<0.0.2)
-Requires-Dist: neo4j (>=5.28.1,<6.0.0)
-Requires-Dist: pydantic (>=2.10.6,<3.0.0)
-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: tiktoken (>=0.9.0,<0.10.0)
-Requires-Dist: typer (>=0.15.2,<0.16.0)
+Author-email: Adam Pingel <oss@pingel.org>
+License-Expression: Apache-2.0
+Requires-Python: >=3.11
 Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: aisuite>=0.1.10
+Requires-Dist: docstring_parser>=0.16
+Requires-Dist: pydantic>=2.10.6
+Requires-Dist: rich>=13.9.4
+Requires-Dist: slack_sdk>=3.35.0
+Provides-Extra: test
+Requires-Dist: pytest<9,>=8; extra == "test"
+Requires-Dist: pytest-cov<6,>=5; extra == "test"
+Requires-Dist: typer>=0.15.2; extra == "test"
+Requires-Dist: click>=8.2.0; extra == "test"
+Requires-Dist: pytest>=8.3.5; extra == "test"
+Requires-Dist: neo4j>=5.28.1; extra == "test"
+Requires-Dist: gofannon>=0.25.13; extra == "test"
+Requires-Dist: lapidarist>=0.0.2; extra == "test"
+Dynamic: license-file
 
 # Proscenium
 
@@ -46,4 +38,3 @@ Currently, proscenium development prioritizes support for domains where the crea
 See the [website](https://the-ai-alliance.github.io/proscenium/) for quickstart info, goals, and other links.
 
 To find the Proscenium community, see the [discussions](https://github.com/The-AI-Alliance/proscenium/discussions)
-

{proscenium-0.0.9.dist-info → proscenium-0.0.11.dist-info}/RECORD

@@ -7,19 +7,20 @@ proscenium/interfaces/__init__.py,sha256=nDWNd6_TSf4vDQuHVBoAf4QfZCB3ZUFQ0M7Xvif
 proscenium/interfaces/slack.py,sha256=wjkdhqAHgISxdNG5Pwo-8_QIyQf76X7b-0WwZfNabI8,10600
 proscenium/patterns/__init__.py,sha256=nDWNd6_TSf4vDQuHVBoAf4QfZCB3ZUFQ0M7XvifNJ-g,78
 proscenium/patterns/graph_rag.py,sha256=1HH1xdlAA6ypvYdP4dWFm-KXrGPUmm0T4qIdAU8mgvE,1763
-proscenium/patterns/rag.py,sha256=loeNWtcDkGjo2UpEIt6B0uBxRKkUtlOHdeCKNjHcupg,1495
+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=Y1T49OfAV7K8p0DMzE4aVqtkgVfjUqb6IeOazzdYGow,5071
-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/verbs/display.py,sha256=hHFmktyJtjYLi4I1-8HUfmsuoMTIxc6JFfczASBsCbI,260
 proscenium/verbs/invoke.py,sha256=-Bk7Pp0EEwRTS0MJUlViZeUNo8wxnDKJj5q78KU4CdM,339
 proscenium/verbs/remember.py,sha256=Hh9BDRAYf7MGeMD4MzU73p6Q28KrSiLWPx4GjTW1amQ,296
-proscenium-0.0.9.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-proscenium-0.0.9.dist-info/METADATA,sha256=OEKKTo6yw_4rnR404Y7pdzuXhJw8yvNcQnzZUxN6juM,2530
-proscenium-0.0.9.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-proscenium-0.0.9.dist-info/entry_points.txt,sha256=Q05DVkPq_SjgD8mFN6bG5ae2r_UbsqKCdy2kDAtHYGU,57
-proscenium-0.0.9.dist-info/RECORD,,
+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.11.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+proscenium-0.0.11.dist-info/METADATA,sha256=wVnK3-0GveteN4CZ4JbmD2ytWZoBu7h8JoZ28ZUdl4w,2102
+proscenium-0.0.11.dist-info/WHEEL,sha256=Nw36Djuh_5VDukK0H78QzOX-_FQEo6V37m3nkm96gtU,91
+proscenium-0.0.11.dist-info/entry_points.txt,sha256=bbrDWv4WKIx9UROGXcyH0eo40F8SNpDQKvXxoE8BTHI,58
+proscenium-0.0.11.dist-info/top_level.txt,sha256=aiTgQy73e21wnSSxM5iICXmXT3zldN8cFKeXpvJ5Xx4,11
+proscenium-0.0.11.dist-info/RECORD,,

{proscenium-0.0.9.dist-info → proscenium-0.0.11.dist-info}/WHEEL

@@ -1,4 +1,5 @@
 Wheel-Version: 1.0
-Generator: poetry-core 2.1.3
+Generator: setuptools (80.7.1)
 Root-Is-Purelib: true
 Tag: py3-none-any
+

proscenium-0.0.11.dist-info/entry_points.txt

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

proscenium-0.0.11.dist-info/top_level.txt

@@ -0,0 +1 @@
+proscenium

proscenium-0.0.9.dist-info/entry_points.txt

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