hyperpocket-llamaindex 0.2.1__tar.gz → 0.3.1__tar.gz

{hyperpocket_llamaindex-0.2.1 → hyperpocket_llamaindex-0.3.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: hyperpocket-llamaindex
-Version: 0.2.1
+Version: 0.3.1
 Author-email: Hyperpocket Team <hyperpocket@vessl.ai>
 Requires-Python: >=3.10
 Requires-Dist: hyperpocket>=0.0.3

{hyperpocket_llamaindex-0.2.1 → hyperpocket_llamaindex-0.3.1}/pyproject.toml

@@ -1,7 +1,7 @@
 
 [project]
 name = "hyperpocket-llamaindex"
-version = "0.2.1"
+version = "0.3.1"
 description = ""
 authors = [{ name = "Hyperpocket Team", email = "hyperpocket@vessl.ai" }]
 requires-python = ">=3.10"

{hyperpocket_llamaindex-0.2.1 → hyperpocket_llamaindex-0.3.1}/tests/test_pocket_llamaindex_no_profile.py

@@ -1,13 +1,13 @@
 import ast
 import json
+import os
 from unittest.async_case import IsolatedAsyncioTestCase
 
-from hyperpocket.config import config, secret
-from hyperpocket.tool import from_git
 from llama_index.agent.openai import OpenAIAgent
 from llama_index.llms.openai import OpenAI
 from pydantic import BaseModel
 
+from hyperpocket.config import config
 from hyperpocket_llamaindex import PocketLlamaindex
 
 
@@ -21,18 +21,14 @@ class TestPocketLlamaindexNoProfile(IsolatedAsyncioTestCase):
 
         self.pocket = PocketLlamaindex(
             tools=[
-                from_git(
-                    "https://github.com/vessl-ai/hyperawesometools",
-                    "main",
-                    "managed-tools/none/simple-echo-tool",
-                ),
+                "https://github.com/vessl-ai/hyperpocket/main/tree/tools/none/simple-echo-tool",
                 self.add,
                 self.sub_pydantic_args,
             ],
             use_profile=False,
         )
 
-        self.llm = OpenAI(model="gpt-4o", api_key=secret["OPENAI_API_KEY"])
+        self.llm = OpenAI(model="gpt-4o", api_key=os.getenv("OPENAI_API_KEY"))
 
     async def asyncTearDown(self):
         self.pocket._teardown_server()
@@ -41,7 +37,7 @@ class TestPocketLlamaindexNoProfile(IsolatedAsyncioTestCase):
         # given
         agent = OpenAIAgent.from_tools(
             tools=self.pocket.get_tools(),
-            llm=OpenAI(model="gpt-4o", api_key=secret["OPENAI_API_KEY"]),
+            llm=OpenAI(model="gpt-4o", api_key=os.getenv("OPENAI_API_KEY")),
             verbose=True,
         )
 

{hyperpocket_llamaindex-0.2.1 → hyperpocket_llamaindex-0.3.1}/tests/test_pocket_llamaindex_use_profile.py

@@ -1,13 +1,13 @@
 import ast
 import json
+import os
 from unittest.async_case import IsolatedAsyncioTestCase
 
-from hyperpocket.config import config, secret
-from hyperpocket.tool import from_git
 from llama_index.agent.openai import OpenAIAgent
 from llama_index.llms.openai import OpenAI
 from pydantic import BaseModel
 
+from hyperpocket.config import config
 from hyperpocket_llamaindex import PocketLlamaindex
 
 
@@ -21,18 +21,14 @@ class TestPocketLlamaindexUseProfile(IsolatedAsyncioTestCase):
 
         self.pocket = PocketLlamaindex(
             tools=[
-                from_git(
-                    "https://github.com/vessl-ai/hyperawesometools",
-                    "main",
-                    "managed-tools/none/simple-echo-tool",
-                ),
+                "https://github.com/vessl-ai/hyperpocket/main/tree/tools/none/simple-echo-tool",
                 self.add,
                 self.sub_pydantic_args,
             ],
             use_profile=True,
         )
 
-        self.llm = OpenAI(model="gpt-4o", api_key=secret["OPENAI_API_KEY"])
+        self.llm = OpenAI(model="gpt-4o", api_key=os.getenv("OPENAI_API_KEY"))
 
     async def asyncTearDown(self):
         self.pocket._teardown_server()
@@ -41,7 +37,7 @@ class TestPocketLlamaindexUseProfile(IsolatedAsyncioTestCase):
         # given
         agent = OpenAIAgent.from_tools(
             tools=self.pocket.get_tools(),
-            llm=OpenAI(model="gpt-4o", api_key=secret["OPENAI_API_KEY"]),
+            llm=OpenAI(model="gpt-4o", api_key=os.getenv("OPENAI_API_KEY")),
             verbose=True,
         )