pinetext 0.1.0__tar.gz → 0.1.2__tar.gz

pinetext-0.1.2/.devcontainer/Dockerfilei

@@ -0,0 +1 @@
+FROM ghcr.io/astral-sh/uv:python3.13-bookworm-slim

pinetext-0.1.2/.devcontainer/devcontainer.json

@@ -0,0 +1,13 @@
+{
+  "build": {
+    "dockerfile": "Dockerfile"
+  },
+  "customizations": {
+    "vscode": {
+      "extensions": [
+        "ms-python.python",
+        "ms-python.vscode-pylance"
+      ]
+    }
+  }
+}

pinetext-0.1.2/.dockerignore

@@ -0,0 +1,7 @@
+*
+!Dockerfile
+!pyproject.toml
+!README.md
+!uv.lock
+!src/
+!src/**

pinetext-0.1.2/.env.example

@@ -0,0 +1,6 @@
+pinetext_pinecone__api_key=your-api-key
+pinetext_pinecone__assistant=assistant-name
+pinetext_pinecone__data_dir=data
+pinetext_pinecone__model=o4-mini
+pinetext_wandb__api_key=wandb-api-key
+pinetext_wandb__project=pinetext

pinetext-0.1.2/.github/dependabot.yml

@@ -0,0 +1,10 @@
+version: 2
+updates:
+  - package-ecosystem: "uv"
+    directory: "/"
+    schedule:
+      interval: "weekly"
+    groups:
+      all:
+        patterns:
+          - "*"

pinetext-0.1.2/.github/workflows/prod.yml

@@ -0,0 +1,38 @@
+# prod.yml
+
+name: prod
+
+on:
+  release:
+    types: [created]
+
+jobs:
+  publish:
+    permissions:
+      contents: read
+      packages: write
+    runs-on: ubuntu-latest
+
+    steps:
+      - uses: actions/checkout@v4
+      - uses: docker/login-action@v3
+        with:
+          registry: ghcr.io
+          username: ${{ github.actor }}
+          password: ${{ secrets.GITHUB_TOKEN }}
+      - name: Metadata
+        id: meta
+        uses: docker/metadata-action@v5
+        with:
+          images: ghcr.io/ezhuk/pinetext
+          tags: |
+            type=semver,pattern={{version}}
+            type=semver,pattern=latest
+      - uses: docker/setup-buildx-action@v3
+      - uses: docker/build-push-action@v6
+        with:
+          cache-from: type=gha
+          cache-to: type=gha,mode=max
+          context: .
+          push: true
+          tags: ${{ steps.meta.outputs.tags }}

{pinetext-0.1.0 → pinetext-0.1.2}/.github/workflows/test.yml

@@ -25,7 +25,7 @@ jobs:
       - run: |
           uv sync
           uv run pre-commit run --all-files
-          uv run pytest --cov=bacnet_mcp --cov-report=xml
+          uv run pytest --cov=pinetext --cov-report=xml
       - uses: codecov/codecov-action@v5
         with:
           use_oidc: true

pinetext-0.1.2/Dockerfile

@@ -0,0 +1,20 @@
+FROM ghcr.io/astral-sh/uv:python3.13-bookworm-slim
+
+WORKDIR /app
+
+ENV UV_COMPILE_BYTECODE=1 \
+    UV_LINK_MODE=copy
+
+COPY pyproject.toml README.md uv.lock ./
+
+RUN --mount=type=cache,target=/root/.cache/uv \
+    uv sync --locked --no-install-project --no-dev
+
+COPY src /app/src
+
+RUN --mount=type=cache,target=/root/.cache/uv \
+    uv sync --locked --no-dev
+
+ENV PATH="/app/.venv/bin:$PATH"
+
+ENTRYPOINT ["pinetext"]

pinetext-0.1.2/PKG-INFO

@@ -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/README.md

@@ -0,0 +1,80 @@
+## 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/docker-compose.yml

@@ -0,0 +1,16 @@
+services:
+  pinetext:
+    build: .
+    container_name: pinetext
+    restart: unless-stopped
+    env_file:
+      - .env
+    environment:
+      - pinetext_pinecone__api_key=${pinetext_pinecone__api_key:-}
+      - pinetext_pinecone__assistant=${pinetext_pinecone__assistant:-assistant-name}
+      - pinetext_pinecone__model=${pinetext_pinecone__model:-o4-mini}
+      - pinetext_pinecone__data_dir=/app/data
+    volumes:
+      - ${pinetext_pinecone__data_dir:-./data}:/app/data
+    stdin_open: true
+    tty: true

pinetext-0.1.2/docs/docs.json

@@ -0,0 +1,32 @@
+{
+  "$schema": "https://mintlify.com/docs.json",
+  "colors": {
+    "primary": "#16A34A",
+    "light": "#07C983",
+    "dark": "#15803D"
+  },
+  "description": "",
+  "icons": {
+    "library": "lucide"
+  },
+  "name": "Docs",
+  "navigation": {
+    "dropdowns": [
+      {
+        "dropdown": "PineText Assistant",
+        "description": "Reason over documents",
+        "icon": "file-search",
+        "groups": [
+          {
+            "group": "Getting Started",
+            "pages": [
+              "pinetext/introduction",
+              "pinetext/installation"
+            ]
+          }
+        ]
+      }
+    ]
+  },
+  "theme": "maple"
+}

pinetext-0.1.2/docs/pinetext/installation.mdx

@@ -0,0 +1,53 @@
+---
+title: Installation
+icon: terminal
+---
+
+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/modbus-mcp/blob/main/docs/modbus-mcp/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
+```

pinetext-0.1.2/docs/pinetext/introduction.mdx

@@ -0,0 +1,6 @@
+---
+title: Introduction
+icon: book-open
+---
+
+PineText is 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.

{pinetext-0.1.0 → pinetext-0.1.2}/pyproject.toml

@@ -1,13 +1,16 @@
 [project]
 name = "pinetext"
-version = "0.1.0"
+version = "0.1.2"
 description = "PineText"
 readme = "README.md"
 requires-python = ">=3.13"
 dependencies = [
     "pinecone>=7.3.0",
+    "pinecone-plugin-assistant>=1.7.0",
     "pydantic-settings>=2.10.1",
     "typer>=0.16.0",
+    "wandb>=0.21.1",
+    "weave>=0.51.59",
 ]
 
 [project.scripts]

pinetext-0.1.2/src/pinetext/client.py

@@ -0,0 +1,59 @@
+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
+
+
+settings = Settings()
+
+
+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):
+        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-0.1.2/src/pinetext/settings.py

@@ -0,0 +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: 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/tests/client_test.py

@@ -0,0 +1,25 @@
+import builtins
+
+
+def test_get_or_create_assistant(pinetext):
+    assistant = pinetext.get_or_create_assistant("foo")
+    assert assistant is pinetext.pinecone.assistant
+
+
+def test_upload_files(pinetext, tmp_path):
+    data = tmp_path / "data"
+    data.mkdir()
+    (data / "test.txt").write_text("TEST")
+    pinetext.upload_files(str(data))
+    assert "test.txt" in pinetext.pinecone.assistant.list_files()
+
+
+def test_chat(pinetext):
+    resp = pinetext.chat("This is a test", model="test-model")
+    assert resp == "Test"
+
+
+def test_run(pinetext, monkeypatch):
+    monkeypatch.setattr(builtins, "input", lambda prompt="": next(iter(["exit"])))
+    res = pinetext.run()
+    assert res is None

pinetext-0.1.2/tests/conftest.py

@@ -0,0 +1,60 @@
+import pytest
+
+from pathlib import Path
+
+import pinetext.client as client_mod
+from pinetext.client import PineText
+
+
+@pytest.fixture
+def cli(monkeypatch):
+    def dummy_run(self):
+        return
+
+    monkeypatch.setattr(
+        "pinetext.client.PineText.run",
+        dummy_run,
+    )
+
+
+@pytest.fixture
+def pinetext(monkeypatch):
+    class DummyAssistant:
+        def __init__(self):
+            self.files = []
+
+        def create_assistant(self, assistant_name: str, instructions: str = None):
+            return self
+
+        def describe_assistant(self, assistant_name: str):
+            return self
+
+        def list_files(self):
+            return self.files
+
+        def upload_file(self, file_path: str, metadata=None, timeout=None):
+            self.files.append(Path(file_path).name)
+
+        def chat(self, messages, model=None):
+            class Message:
+                def __init__(self, content):
+                    self.content = content
+
+            class Response:
+                def __init__(self, content):
+                    self.message = Message(content)
+
+            return Response("Test")
+
+    assistant = DummyAssistant()
+
+    class DummyPinecone:
+        def __init__(self, api_key):
+            self.assistant = assistant
+
+    monkeypatch.setattr(client_mod, "Pinecone", DummyPinecone)
+
+    client = PineText()
+    client.assistant = assistant
+    client.pinecone = DummyPinecone(None)
+    return client