pinetext 0.1.0__py3-none-any.whl → 0.1.2__py3-none-any.whl

pinetext/client.py

@@ -1,3 +1,11 @@
+import wandb
+import weave
+
+from pathlib import Path
+
+from pinecone import Pinecone
+from pinecone_plugins.assistant.models.chat import Message
+
 from pinetext.settings import Settings
 
 
@@ -8,5 +16,44 @@ class PineText:
     def __init__(self):
         pass
 
+    def get_or_create_assistant(self, name: str):
+        try:
+            return self.pinecone.assistant.describe_assistant(assistant_name=name)
+        except Exception:
+            return self.pinecone.assistant.create_assistant(assistant_name=name)
+
+    def upload_files(self, path: str):
+        folder = Path(path)
+        if folder.is_dir():
+            uploaded = [x.name for x in self.assistant.list_files()]
+            for x in sorted(folder.iterdir()):
+                if x.name not in uploaded:
+                    self.assistant.upload_file(
+                        file_path=str(x.resolve()),
+                        metadata={
+                            "filename": x.name,
+                            "extension": x.suffix.lower().lstrip("."),
+                        },
+                        timeout=None,
+                    )
+
+    @weave.op()
+    def chat(self, text: str, model: str):
+        msg = Message(role="user", content=text)
+        resp = self.assistant.chat(messages=[msg], model=model)
+        return resp.message.content
+
     def run(self):
-        print("OK")
+        if settings.wandb.api_key:
+            wandb.login(key=settings.wandb.api_key)
+            weave.init(settings.wandb.project)
+        self.pinecone = Pinecone(api_key=settings.pinecone.api_key)
+        self.assistant = self.get_or_create_assistant(settings.pinecone.assistant)
+        self.upload_files(settings.pinecone.data_dir)
+
+        while True:
+            text = input("> ").strip()
+            if text.lower() in ("exit", "quit"):
+                break
+            res = self.chat(text, settings.pinecone.model)
+            print(res)

pinetext/settings.py

@@ -1,10 +1,25 @@
+from pydantic import BaseModel
 from pydantic_settings import BaseSettings, SettingsConfigDict
 
 
+class Pinecone(BaseModel):
+    api_key: str | None = None
+    assistant: str | None = "test-assistant"
+    data_dir: str | None = "data"
+    model: str | None = "o4-mini"
+
+
+class WandB(BaseModel):
+    api_key: str | None = None
+    project: str | None = "pinetext"
+
+
 class Settings(BaseSettings):
-    pinecone_key: str | None = None
+    pinecone: Pinecone = Pinecone()
+    wandb: WandB = WandB()
     model_config = SettingsConfigDict(
         env_file=".env",
         env_file_encoding="utf-8",
         env_nested_delimiter="__",
+        env_prefix="PINETEXT_",
     )

pinetext-0.1.2.dist-info/METADATA

@@ -0,0 +1,94 @@
+Metadata-Version: 2.4
+Name: pinetext
+Version: 0.1.2
+Summary: PineText
+License-File: LICENSE
+Requires-Python: >=3.13
+Requires-Dist: pinecone-plugin-assistant>=1.7.0
+Requires-Dist: pinecone>=7.3.0
+Requires-Dist: pydantic-settings>=2.10.1
+Requires-Dist: typer>=0.16.0
+Requires-Dist: wandb>=0.21.1
+Requires-Dist: weave>=0.51.59
+Description-Content-Type: text/markdown
+
+## PineText
+
+[![test](https://github.com/ezhuk/pinetext/actions/workflows/test.yml/badge.svg)](https://github.com/ezhuk/pinetext/actions/workflows/test.yml)
+[![codecov](https://codecov.io/github/ezhuk/pinetext/graph/badge.svg?token=0YJASFE5OM)](https://codecov.io/github/ezhuk/pinetext)
+[![PyPI - Version](https://img.shields.io/pypi/v/pinetext.svg)](https://pypi.org/p/pinetext)
+
+A lightweight assistant built using [Pinecone](https://docs.pinecone.io/guides/assistant/overview) that helps create RAG-based chat applications for reasoning over documents, retrieving relevant context, and providing grounded answers.
+
+## Getting Started
+
+Use [uv](https://github.com/astral-sh/uv) to add and manage PineText as a dependency in your project, or install it directly via `uv pip install` or `pip install`. See the [Installation](https://github.com/ezhuk/pinetext/blob/main/docs/pinetext/installation.mdx) section of the documentation for full installation instructions and more details.
+
+```bash
+uv add pinetext
+```
+
+It can be embedded in and run directly from your application.
+
+```python
+# app.py
+from pinetext import PineText
+
+def main():
+    pt = PineText()
+    pt.run()
+```
+
+It can also be launched from the command line using the provided `CLI` without modifying the source code.
+
+```
+pinetext
+```
+
+Or in an ephemeral, isolated environment using `uvx`. Check out the [Using tools](https://docs.astral.sh/uv/guides/tools/) guide for more details.
+
+```bash
+uvx pinetext
+```
+
+## Configuration
+
+Place documents in the `data` folder and make sure to set `PINECONE_API_KEY` and the assistant name before starting PineText.
+
+```bash
+export PINETEXT_PINECONE__API_KEY=your-api-key
+export PINETEXT_PINECONE__ASSISTANT=assistant-name
+export PINETEXT_PINECONE__DATA_DIR=data
+export PINETEXT_PINECONE__MODEL=o4-mini
+```
+
+These settings can also be specified in a `.env` file in the working directory.
+
+```text
+pinetext_pinecone__api_key=your-api-key
+pinetext_pinecone__assistant=assistant-name
+pinetext_pinecone__data_dir=data
+pinetext_pinecone__model=o4-mini
+```
+
+## Docker
+
+The PineText CLI can be deployed as a Docker container as follows:
+
+```bash
+docker run -it \
+  --name pinetext \
+  --env-file .env \
+  -v $(pwd)/data:/app/data
+  ghcr.io/ezhuk/pinetext:latest
+```
+
+Or using Docker Compose:
+
+```bash
+docker compose up
+```
+
+## License
+
+The server is licensed under the [MIT License](https://github.com/ezhuk/pinetext?tab=MIT-1-ov-file).

pinetext-0.1.2.dist-info/RECORD

@@ -0,0 +1,9 @@
+pinetext/__init__.py,sha256=VQZQcYgadJGizofgf7Qe6vPlC8zIiv_kTiQC4Ftzfac,136
+pinetext/cli.py,sha256=ensHQ7RQvsZG7PIvSEVsRLJEvgPOf5ZrVDRUi2ZsJJM,189
+pinetext/client.py,sha256=hYKhaefZ4L6PUDvPxwicjnztlAkgVLIZxM8sr5he_18,1892
+pinetext/settings.py,sha256=B4-gtpFlaz-M1Zf5RXC5MK2gh8faLLHUxtO7wWPCgsI,628
+pinetext-0.1.2.dist-info/METADATA,sha256=KxtSrVYTVYEQQP6F9DuGR4d_9pMh5kTqM_L3iduXads,2798
+pinetext-0.1.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+pinetext-0.1.2.dist-info/entry_points.txt,sha256=kV_mlYLNZRAerrjTMaKHwpQoPAejaJ7Crjvpck5D3KQ,46
+pinetext-0.1.2.dist-info/licenses/LICENSE,sha256=uSsFh4jumQQchhdeV61a1YhWRAp1GouNAoEduU_93kU,1068
+pinetext-0.1.2.dist-info/RECORD,,

pinetext-0.1.0.dist-info/METADATA

@@ -1,15 +0,0 @@
-Metadata-Version: 2.4
-Name: pinetext
-Version: 0.1.0
-Summary: PineText
-License-File: LICENSE
-Requires-Python: >=3.13
-Requires-Dist: pinecone>=7.3.0
-Requires-Dist: pydantic-settings>=2.10.1
-Requires-Dist: typer>=0.16.0
-Description-Content-Type: text/markdown
-
-## PineText
-
-[![test](https://github.com/ezhuk/pinetext/actions/workflows/test.yml/badge.svg)](https://github.com/ezhuk/pinetext/actions/workflows/test.yml)
-[![codecov](https://codecov.io/github/ezhuk/pinetext/graph/badge.svg?token=0YJASFE5OM)](https://codecov.io/github/ezhuk/pinetext)

pinetext-0.1.0.dist-info/RECORD

@@ -1,9 +0,0 @@
-pinetext/__init__.py,sha256=VQZQcYgadJGizofgf7Qe6vPlC8zIiv_kTiQC4Ftzfac,136
-pinetext/cli.py,sha256=ensHQ7RQvsZG7PIvSEVsRLJEvgPOf5ZrVDRUi2ZsJJM,189
-pinetext/client.py,sha256=AyKifEdono9_J3hzaqMI55tjMprEZAB-yieTTnISJU4,158
-pinetext/settings.py,sha256=v1Twv_5MtgOfkYHamnp9ZKa8Jv2hxALJLweAzLBWkqw,271
-pinetext-0.1.0.dist-info/METADATA,sha256=mBsLrlHrtSsfv-eh3Xutdjs5gahimHcwkzUK2rSgVnc,545
-pinetext-0.1.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-pinetext-0.1.0.dist-info/entry_points.txt,sha256=kV_mlYLNZRAerrjTMaKHwpQoPAejaJ7Crjvpck5D3KQ,46
-pinetext-0.1.0.dist-info/licenses/LICENSE,sha256=uSsFh4jumQQchhdeV61a1YhWRAp1GouNAoEduU_93kU,1068
-pinetext-0.1.0.dist-info/RECORD,,