hyperpocket-langgraph 0.2.0__tar.gz → 0.3.0__tar.gz

Sign up to get free protection for your applications and to get access to all the features.
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: hyperpocket-langgraph
3
- Version: 0.2.0
3
+ Version: 0.3.0
4
4
  Author-email: Hyperpocket Team <hyperpocket@vessl.ai>
5
5
  Requires-Python: >=3.10
6
6
  Requires-Dist: hyperpocket>=0.0.3
@@ -1,7 +1,7 @@
1
1
 
2
2
  [project]
3
3
  name = "hyperpocket-langgraph"
4
- version = "0.2.0"
4
+ version = "0.3.0"
5
5
  description = ""
6
6
  authors = [{ name = "Hyperpocket Team", email = "hyperpocket@vessl.ai" }]
7
7
  requires-python = ">=3.10"
@@ -1,14 +1,14 @@
1
1
  import ast
2
+ import os
2
3
  from unittest.async_case import IsolatedAsyncioTestCase
3
4
 
4
- from hyperpocket.config import config, secret
5
- from hyperpocket.tool import from_git
6
5
  from langchain_openai import ChatOpenAI
7
6
  from langgraph.constants import END, START
8
7
  from langgraph.graph import MessagesState, StateGraph
9
8
  from langgraph.prebuilt import tools_condition
10
9
  from pydantic import BaseModel
11
10
 
11
+ from hyperpocket.config import config
12
12
  from hyperpocket_langgraph import PocketLanggraph
13
13
 
14
14
 
@@ -22,11 +22,7 @@ class TestPocketLanggraphNoProfile(IsolatedAsyncioTestCase):
22
22
 
23
23
  self.pocket = PocketLanggraph(
24
24
  tools=[
25
- from_git(
26
- "https://github.com/vessl-ai/hyperawesometools",
27
- "main",
28
- "managed-tools/none/simple-echo-tool",
29
- ),
25
+ "https://github.com/vessl-ai/hyperpocket/main/tree/tools/none/simple-echo-tool",
30
26
  self.add,
31
27
  self.sub_pydantic_args,
32
28
  ],
@@ -35,7 +31,7 @@ class TestPocketLanggraphNoProfile(IsolatedAsyncioTestCase):
35
31
  tools = self.pocket.get_tools()
36
32
  tool_node = self.pocket.get_tool_node()
37
33
  self.llm = ChatOpenAI(
38
- model="gpt-4o", api_key=secret["OPENAI_API_KEY"]
34
+ model="gpt-4o", api_key=os.getenv("OPENAI_API_KEY")
39
35
  ).bind_tools(tools=tools)
40
36
 
41
37
  def chatbot(state: MessagesState):
@@ -1,7 +1,8 @@
1
1
  import ast
2
+ import os
2
3
  from unittest.async_case import IsolatedAsyncioTestCase
3
4
 
4
- from hyperpocket.config import config, secret
5
+ from hyperpocket.config import config
5
6
  from hyperpocket.tool import from_git
6
7
  from langchain_openai import ChatOpenAI
7
8
  from langgraph.constants import END, START
@@ -22,11 +23,7 @@ class TestPocketLanggraphUseProfile(IsolatedAsyncioTestCase):
22
23
 
23
24
  self.pocket = PocketLanggraph(
24
25
  tools=[
25
- from_git(
26
- "https://github.com/vessl-ai/hyperawesometools",
27
- "main",
28
- "managed-tools/none/simple-echo-tool",
29
- ),
26
+ "https://github.com/vessl-ai/hyperpocket/main/tree/tools/none/simple-echo-tool",
30
27
  self.add,
31
28
  self.sub_pydantic_args,
32
29
  ],
@@ -35,7 +32,7 @@ class TestPocketLanggraphUseProfile(IsolatedAsyncioTestCase):
35
32
  tools = self.pocket.get_tools()
36
33
  tool_node = self.pocket.get_tool_node()
37
34
  self.llm = ChatOpenAI(
38
- model="gpt-4o", api_key=secret["OPENAI_API_KEY"]
35
+ model="gpt-4o", api_key=os.getenv("OPENAI_API_KEY")
39
36
  ).bind_tools(tools=tools)
40
37
 
41
38
  def chatbot(state: MessagesState):