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

solana_agent/ai.py

@@ -215,8 +215,7 @@ class AI:
             else:
                 uid = uuid.uuid4().hex
                 self._vector_store = self._client.beta.vector_stores.create(
-                    name=uid
-                )
+                    name=uid)
                 self._database.save_vector_store_id(self._vector_store.id)
             self._client.beta.assistants.update(
                 assistant_id=self._assistant_id,
@@ -335,24 +334,14 @@ class AI:
 
     def add_file(
         self,
+        filename: str,
         file_stream: bytes,
-        file_extension: Literal[
-            "doc", "docx", "json", "md", "pdf", "pptx", "tex", "txt"
-        ] = "pdf",
     ) -> 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
-            file_extension (Literal, optional): File type extension. Defaults to "pdf"
-                Supported formats:
-                - doc, docx: Word documents
-                - json: JSON files
-                - md: Markdown files
-                - pdf: PDF documents
-                - pptx: PowerPoint presentations
-                - tex: LaTeX files
-                - txt: Plain text files
 
         Returns:
             Literal["in_progress", "completed", "cancelled", "failed"]: Status of file processing
@@ -360,7 +349,7 @@ class AI:
         Example:
             ```python
             with open('document.pdf', 'rb') as f:
-                status = ai.add_file(f.read(), file_extension="pdf")
+                status = ai.add_file(f.filename, f.read())
                 if status == "completed":
                     print("File processed successfully")
             ```
@@ -374,7 +363,7 @@ class AI:
         """
         vector_store_id = self._database.get_vector_store_id()
         file = self._client.files.create(
-            file=(f"file.{file_extension}", file_stream), purpose="assistants"
+            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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: solana-agent
-Version: 1.3.0
+Version: 1.3.1
 Summary: Build self-learning AI Agents
 License: MIT
 Keywords: ai,openai,ai agents

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.3.0.dist-info/RECORD

@@ -1,6 +0,0 @@
-solana_agent/__init__.py,sha256=zpfnWqANd3OHGWm7NCF5Y6m01BWG4NkNk8SK9Ex48nA,18
-solana_agent/ai.py,sha256=o6ztao5oFuzatYIkYxGlQSf16JYgQeIaUtCwL15ojoU,41942
-solana_agent-1.3.0.dist-info/LICENSE,sha256=BnSRc-NSFuyF2s496l_4EyrwAP6YimvxWcjPiJ0J7g4,1057
-solana_agent-1.3.0.dist-info/METADATA,sha256=bqeaN3lk98Fd1UWZmuZaH_HiEgNo11Q0K0GmNhbTgoY,4694
-solana_agent-1.3.0.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
-solana_agent-1.3.0.dist-info/RECORD,,