solana-agent 1.2.0__py3-none-any.whl → 1.3.1__py3-none-any.whl

solana_agent/ai.py

@@ -49,6 +49,8 @@ class MongoDatabase:
         self._threads = self.db["threads"]
         self.messages = self.db["messages"]
         self.kb = self.db["kb"]
+        self.vector_stores = self.db["vector_stores"]
+        self.files = self.db["files"]
 
     def save_thread_id(self, user_id: str, thread_id: str):
         self._threads.insert_one({"thread_id": thread_id, "user_id": user_id})
@@ -76,6 +78,22 @@ class MongoDatabase:
         storage["timestamp"] = datetime.datetime.now(datetime.timezone.utc)
         self.kb.insert_one(storage)
 
+    def get_vector_store_id(self) -> str | None:
+        document = self.vector_stores.find_one()
+        return document["vector_store_id"] if document else None
+
+    def save_vector_store_id(self, vector_store_id: str):
+        self.vector_stores.insert_one({"vector_store_id": vector_store_id})
+
+    def delete_vector_store_id(self, vector_store_id: str):
+        self.vector_stores.delete_one({"vector_store_id": vector_store_id})
+
+    def add_file(self, file_id: str):
+        self.files.insert_one({"file_id": file_id})
+
+    def delete_file(self, file_id: str):
+        self.files.delete_one({"file_id": file_id})
+
 
 class AI:
     def __init__(
@@ -92,6 +110,7 @@ class AI:
         cohere_api_key: str = None,
         cohere_model: Literal["rerank-v3.5"] = "rerank-v3.5",
         code_interpreter: bool = True,
+        file_search: bool = True,
         openai_assistant_model: Literal["gpt-4o-mini",
                                         "gpt-4o"] = "gpt-4o-mini",
         openai_embedding_model: Literal[
@@ -113,6 +132,7 @@ class AI:
             cohere_api_key (str, optional): API key for Cohere search. Defaults to None
             cohere_model (Literal["rerank-v3.5"], optional): Cohere model for reranking. Defaults to "rerank-v3.5"
             code_interpreter (bool, optional): Enable code interpretation. Defaults to True
+            file_search (bool, optional): Enable file search tool. Defaults to True
             openai_assistant_model (Literal["gpt-4o-mini", "gpt-4o"], optional): OpenAI model for assistant. Defaults to "gpt-4o-mini"
             openai_embedding_model (Literal["text-embedding-3-small", "text-embedding-3-large"], optional): OpenAI model for text embedding. Defaults to "text-embedding-3-small"
 
@@ -137,8 +157,20 @@ class AI:
         self._instructions = instructions
         self._openai_assistant_model = openai_assistant_model
         self._openai_embedding_model = openai_embedding_model
-        self._tools = [{"type": "code_interpreter"}
-                       ] if code_interpreter else []
+        self._file_search = file_search
+        if file_search:
+            self._tools = (
+                [
+                    {"type": "code_interpreter"},
+                    {"type": "file_search"},
+                ]
+                if code_interpreter
+                else [{"type": "file_search"}]
+            )
+        else:
+            self._tools = [{"type": "code_interpreter"}
+                           ] if code_interpreter else []
+
         self._tool_handlers = {}
         self._assistant_id = None
         self._database: MongoDatabase = database
@@ -174,7 +206,23 @@ class AI:
                 model=self._openai_assistant_model,
             ).id
             self._database.delete_all_threads()
-
+        if self._file_search:
+            vectore_store_id = self._database.get_vector_store_id()
+            if vectore_store_id:
+                self._vector_store = self._client.beta.vector_stores.retrieve(
+                    vector_store_id=vectore_store_id
+                )
+            else:
+                uid = uuid.uuid4().hex
+                self._vector_store = self._client.beta.vector_stores.create(
+                    name=uid)
+                self._database.save_vector_store_id(self._vector_store.id)
+            self._client.beta.assistants.update(
+                assistant_id=self._assistant_id,
+                tool_resources={
+                    "file_search": {"vector_store_ids": [self._vector_store.id]}
+                },
+            )
         return self
 
     async def __aexit__(self, exc_type, exc_val, exc_tb):
@@ -223,6 +271,106 @@ class AI:
         )
         return run.status
 
+    def delete_vector_store_and_files(self):
+        """Delete the OpenAI vector store and files.
+
+        Example:
+            ```python
+            ai.delete_vector_store()
+            ```
+
+        Note:
+            - Requires file_search=True in AI initialization
+            - Deletes the vector store and all associated files
+        """
+        vector_store_id = self._database.get_vector_store_id()
+        if vector_store_id:
+            self._client.beta.vector_stores.delete(vector_store_id)
+            self._database.delete_vector_store_id(vector_store_id)
+            for file in self._database.files.find().to_list():
+                self._client.files.delete(file["file_id"])
+                self._database.delete_file(file["file_id"])
+
+    def max_files(self) -> bool:
+        """Check if the OpenAI vector store has reached its maximum file capacity.
+
+        Returns:
+            bool: True if file count is at maximum (10,000), False otherwise
+
+        Example:
+            ```python
+            if ai.max_files():
+                print("Vector store is full")
+            else:
+                print("Can still add more files")
+            ```
+
+        Note:
+            - Requires file_search=True in AI initialization
+            - OpenAI vector stores have a 10,000 file limit
+            - Returns False if vector store is not configured
+        """
+        self._vector_store.file_counts.completed == 10000
+
+    def file_count(self) -> int:
+        """Get the total number of files processed in the OpenAI vector store.
+
+        Returns:
+            int: Number of successfully processed files in the vector store
+
+        Example:
+            ```python
+            count = ai.file_count()
+            print(f"Processed {count} files")
+            # Returns: "Processed 5 files"
+            ```
+
+        Note:
+            - Requires file_search=True in AI initialization
+            - Only counts successfully processed files
+            - Returns 0 if vector store is not configured
+        """
+        self._vector_store.file_counts.completed
+
+    def add_file(
+        self,
+        filename: str,
+        file_stream: bytes,
+    ) -> Literal["in_progress", "completed", "cancelled", "failed"]:
+        """Upload and process a file in the OpenAI vector store.
+
+        Args:
+            filename (str): Name of the file to upload
+            file_stream (bytes): Raw bytes of the file to upload
+
+        Returns:
+            Literal["in_progress", "completed", "cancelled", "failed"]: Status of file processing
+
+        Example:
+            ```python
+            with open('document.pdf', 'rb') as f:
+                status = ai.add_file(f.filename, f.read())
+                if status == "completed":
+                    print("File processed successfully")
+            ```
+
+        Note:
+            - Requires file_search=True in AI initialization
+            - Files are vectorized for semantic search
+            - Maximum file size: 512MB
+            - Maximum 10,000 files per vector store
+            - Processing may take a few seconds to minutes
+        """
+        vector_store_id = self._database.get_vector_store_id()
+        file = self._client.files.create(
+            file=(filename, file_stream), purpose="assistants"
+        )
+        file_batch = self._client.beta.vector_stores.files.create_and_poll(
+            vector_store_id=vector_store_id, file_id=file.id
+        )
+        self._database.add_file(file.id)
+        return file_batch.status
+
     def search_kb(self, query: str, namespace: str = "global", limit: int = 3) -> str:
         """Search Pinecone knowledge base using OpenAI embeddings.
 

{solana_agent-1.2.0.dist-info → solana_agent-1.3.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: solana-agent
-Version: 1.2.0
+Version: 1.3.1
 Summary: Build self-learning AI Agents
 License: MIT
 Keywords: ai,openai,ai agents
@@ -17,7 +17,7 @@ Classifier: Programming Language :: Python :: 3.13
 Classifier: Programming Language :: Python :: 3 :: Only
 Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
 Requires-Dist: cohere (>=5.13.12,<6.0.0)
-Requires-Dist: openai (>=1.61.1,<2.0.0)
+Requires-Dist: openai (>=1.63.2,<2.0.0)
 Requires-Dist: pinecone (>=6.0.1,<7.0.0)
 Requires-Dist: pydantic (>=2.10.6,<3.0.0)
 Requires-Dist: pymongo (>=4.11.1,<5.0.0)
@@ -45,7 +45,7 @@ Unlike traditional AI assistants that forget conversations after each session, S
 - 🎯 **Context-Aware**: Recalls past interactions for more relevant responses
 - 🔄 **Self-Improving**: Builds knowledge and improves reasoning automatically
 - 🧠 **Knowledge Base**: Add domain-specific knowledge for better reasoning
-- 🏢 **Enterprise-Ready**: Scales from personal to organization-wide deployment
+- 🏢 **File Context**: Upload propriety files to be part of the conversation
 - 🛡️ **Secure**: Secure and private memory and data storage 
 
 **Experience Agentic IQ!**
@@ -57,7 +57,7 @@ Unlike traditional AI assistants that forget conversations after each session, S
 - Real-time voice-to-voice conversations
 
 🧠 **Memory System and Extensibility**
-- Advanced AI memory combining conversational context, conversational facts, and parallel tool calling
+- Advanced AI memory combining conversational context, conversational facts, knowledge base, file search, and parallel tool calling
 - Create custom tools for extending the Agent's capabilities like further API integrations
 
 🔍 **Multi-Source Search and Reasoning**
@@ -66,6 +66,7 @@ Unlike traditional AI assistants that forget conversations after each session, S
 - Conversational fact search powered by Zep
 - Conversational message history using MongoDB (on-prem or hosted)
 - Knowledge Base using Pinecone with reranking by Cohere - available globally or user-specific
+- File search using OpenAI
 - Comprehensive reasoning combining multiple data sources
 
 ## Why Choose Solana Agent Over LangChain?
@@ -77,6 +78,7 @@ Unlike traditional AI assistants that forget conversations after each session, S
    - Persistent cross-session knowledge retention
    - Automatic self-learning from conversations
    - Knowledge Base to add domain specific knowledge
+   - File uploads to perform document context search 
 
 🏢 **Enterprise Focus**
    - Production-ready out of the box in a few lines of code

solana_agent-1.3.1.dist-info/RECORD

@@ -0,0 +1,6 @@
+solana_agent/__init__.py,sha256=zpfnWqANd3OHGWm7NCF5Y6m01BWG4NkNk8SK9Ex48nA,18
+solana_agent/ai.py,sha256=RxbyG9UzvFjZSGTZ2SC7t6U0Pfw6uERaAisxGDqLsUw,41457
+solana_agent-1.3.1.dist-info/LICENSE,sha256=BnSRc-NSFuyF2s496l_4EyrwAP6YimvxWcjPiJ0J7g4,1057
+solana_agent-1.3.1.dist-info/METADATA,sha256=I6QyN8qmmeFzpNNlXN4ZquFJ3srdK5FuooKEOqmzTrw,4694
+solana_agent-1.3.1.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
+solana_agent-1.3.1.dist-info/RECORD,,

{solana_agent-1.2.0.dist-info → solana_agent-1.3.1.dist-info}/WHEEL

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

solana_agent-1.2.0.dist-info/RECORD

@@ -1,6 +0,0 @@
-solana_agent/__init__.py,sha256=zpfnWqANd3OHGWm7NCF5Y6m01BWG4NkNk8SK9Ex48nA,18
-solana_agent/ai.py,sha256=Xi3_9bFhF6qsVNsYCSNJ11rXB3aoFsy-UPY2EgUffHI,36023
-solana_agent-1.2.0.dist-info/LICENSE,sha256=BnSRc-NSFuyF2s496l_4EyrwAP6YimvxWcjPiJ0J7g4,1057
-solana_agent-1.2.0.dist-info/METADATA,sha256=ZNkNWKcJ-nJWs7qmQ7693tGVlQ4-nSKz8doRan-QFx8,4587
-solana_agent-1.2.0.dist-info/WHEEL,sha256=7dDg4QLnNKTvwIDR9Ac8jJaAmBC_owJrckbC0jjThyA,88
-solana_agent-1.2.0.dist-info/RECORD,,