pinetext 0.1.1__py3-none-any.whl → 0.1.3__py3-none-any.whl

pinetext/client.py

@@ -1,3 +1,6 @@
+import wandb
+import weave
+
 from pathlib import Path
 
 from pinecone import Pinecone
@@ -34,12 +37,16 @@ class PineText:
                         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)

pinetext/settings.py

@@ -9,8 +9,14 @@ class Pinecone(BaseModel):
     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",

{pinetext-0.1.1.dist-info → pinetext-0.1.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pinetext
-Version: 0.1.1
+Version: 0.1.3
 Summary: PineText
 License-File: LICENSE
 Requires-Python: >=3.13
@@ -8,6 +8,8 @@ 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
@@ -20,7 +22,7 @@ A lightweight assistant built using [Pinecone](https://docs.pinecone.io/guides/a
 
 ## 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/modbus-mcp/blob/main/docs/modbus-mcp/installation.mdx) section of the documentation for full installation instructions and more details.
+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
@@ -81,6 +83,12 @@ docker run -it \
   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.3.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.3.dist-info/METADATA,sha256=b2xnAObh9KTS8ey0LTHpXBtXF2_iwbbv6vPRPCX_z48,2798
+pinetext-0.1.3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+pinetext-0.1.3.dist-info/entry_points.txt,sha256=kV_mlYLNZRAerrjTMaKHwpQoPAejaJ7Crjvpck5D3KQ,46
+pinetext-0.1.3.dist-info/licenses/LICENSE,sha256=uSsFh4jumQQchhdeV61a1YhWRAp1GouNAoEduU_93kU,1068
+pinetext-0.1.3.dist-info/RECORD,,

pinetext-0.1.1.dist-info/RECORD

@@ -1,9 +0,0 @@
-pinetext/__init__.py,sha256=VQZQcYgadJGizofgf7Qe6vPlC8zIiv_kTiQC4Ftzfac,136
-pinetext/cli.py,sha256=ensHQ7RQvsZG7PIvSEVsRLJEvgPOf5ZrVDRUi2ZsJJM,189
-pinetext/client.py,sha256=YSVTNMMverEVqjLliqy0TL8Y-Xo4fHDTgqDq3da0yXk,1715
-pinetext/settings.py,sha256=nkHmg2zilfJA7RDELJB6vaqfuisVygBv5FoMz9Zg0ew,507
-pinetext-0.1.1.dist-info/METADATA,sha256=kxB34o0p3XwyF7iM_fLzkVAwiu-3WH3ru5e0pYYKYNc,2686
-pinetext-0.1.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-pinetext-0.1.1.dist-info/entry_points.txt,sha256=kV_mlYLNZRAerrjTMaKHwpQoPAejaJ7Crjvpck5D3KQ,46
-pinetext-0.1.1.dist-info/licenses/LICENSE,sha256=uSsFh4jumQQchhdeV61a1YhWRAp1GouNAoEduU_93kU,1068
-pinetext-0.1.1.dist-info/RECORD,,