Spaces:
Runtime error
Runtime error
semioz
commited on
Commit
·
69578a4
1
Parent(s):
7f4fddc
ruff
Browse files- agent.py +8 -7
- app.py +2 -1
- pyproject.toml +0 -2
agent.py
CHANGED
@@ -1,4 +1,9 @@
|
|
1 |
import logging
|
|
|
|
|
|
|
|
|
|
|
2 |
from langchain_groq import ChatGroq
|
3 |
from langchain_huggingface import (
|
4 |
ChatHuggingFace,
|
@@ -7,13 +12,9 @@ from langchain_huggingface import (
|
|
7 |
)
|
8 |
from langgraph.graph import START, MessagesState, StateGraph
|
9 |
from langgraph.prebuilt import ToolNode, tools_condition
|
10 |
-
from langchain_community.vectorstores import SupabaseVectorStore
|
11 |
-
from langchain.tools.retriever import create_retriever_tool
|
12 |
from supabase.client import Client, create_client
|
13 |
-
from langchain_core.messages import HumanMessage, SystemMessage
|
14 |
|
15 |
from tools import tools
|
16 |
-
import os
|
17 |
|
18 |
logger = logging.getLogger(__name__)
|
19 |
|
@@ -31,7 +32,7 @@ embeddings = HuggingFaceEmbeddings(
|
|
31 |
)
|
32 |
|
33 |
supabase: Client = create_client(
|
34 |
-
os.environ.get("SUPABASE_URL"),
|
35 |
os.environ.get("SUPABASE_SERVICE_KEY"))
|
36 |
vector_store = SupabaseVectorStore(
|
37 |
client=supabase,
|
@@ -61,14 +62,14 @@ def build_graph(provider: str = "groq"):
|
|
61 |
)
|
62 |
else:
|
63 |
raise ValueError("Invalid provider. Choose 'groq' or 'huggingface'.")
|
64 |
-
llm_with_tools = llm.bind_tools(tools)
|
65 |
|
66 |
# Node
|
67 |
def assistant(state: MessagesState):
|
68 |
"""Assistant node"""
|
69 |
return {"messages": [llm_with_tools.invoke(state["messages"])]}
|
70 |
|
71 |
-
|
72 |
def retriever(state: MessagesState):
|
73 |
"""Retriever node"""
|
74 |
similar_question = vector_store.similarity_search(state["messages"][0].content)
|
|
|
1 |
import logging
|
2 |
+
import os
|
3 |
+
|
4 |
+
from langchain.tools.retriever import create_retriever_tool
|
5 |
+
from langchain_community.vectorstores import SupabaseVectorStore
|
6 |
+
from langchain_core.messages import HumanMessage, SystemMessage
|
7 |
from langchain_groq import ChatGroq
|
8 |
from langchain_huggingface import (
|
9 |
ChatHuggingFace,
|
|
|
12 |
)
|
13 |
from langgraph.graph import START, MessagesState, StateGraph
|
14 |
from langgraph.prebuilt import ToolNode, tools_condition
|
|
|
|
|
15 |
from supabase.client import Client, create_client
|
|
|
16 |
|
17 |
from tools import tools
|
|
|
18 |
|
19 |
logger = logging.getLogger(__name__)
|
20 |
|
|
|
32 |
)
|
33 |
|
34 |
supabase: Client = create_client(
|
35 |
+
os.environ.get("SUPABASE_URL"),
|
36 |
os.environ.get("SUPABASE_SERVICE_KEY"))
|
37 |
vector_store = SupabaseVectorStore(
|
38 |
client=supabase,
|
|
|
62 |
)
|
63 |
else:
|
64 |
raise ValueError("Invalid provider. Choose 'groq' or 'huggingface'.")
|
65 |
+
llm_with_tools = llm.bind_tools(tools)
|
66 |
|
67 |
# Node
|
68 |
def assistant(state: MessagesState):
|
69 |
"""Assistant node"""
|
70 |
return {"messages": [llm_with_tools.invoke(state["messages"])]}
|
71 |
|
72 |
+
|
73 |
def retriever(state: MessagesState):
|
74 |
"""Retriever node"""
|
75 |
similar_question = vector_store.similarity_search(state["messages"][0].content)
|
app.py
CHANGED
@@ -1,9 +1,10 @@
|
|
1 |
import os
|
2 |
|
3 |
-
from langchain_core.messages import HumanMessage
|
4 |
import gradio as gr
|
5 |
import pandas as pd
|
6 |
import requests
|
|
|
|
|
7 |
from agent import build_graph
|
8 |
|
9 |
DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
|
|
|
1 |
import os
|
2 |
|
|
|
3 |
import gradio as gr
|
4 |
import pandas as pd
|
5 |
import requests
|
6 |
+
from langchain_core.messages import HumanMessage
|
7 |
+
|
8 |
from agent import build_graph
|
9 |
|
10 |
DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
|
pyproject.toml
CHANGED
@@ -18,11 +18,9 @@ dependencies = [
|
|
18 |
extend-select = [
|
19 |
"F", # Pyflakes rules
|
20 |
"W", # PyCodeStyle warnings
|
21 |
-
"E", # PyCodeStyle errors
|
22 |
"I", # Sort imports properly
|
23 |
"UP", # Warn if certain things can changed due to newer Python versions
|
24 |
"C4", # Catch incorrect use of comprehensions, dict, list, etc
|
25 |
-
"FA", # Enforce from __future__ import annotations
|
26 |
"ISC", # Good use of string concatenation
|
27 |
"ICN", # Use common import conventions
|
28 |
"RET", # Good return practices
|
|
|
18 |
extend-select = [
|
19 |
"F", # Pyflakes rules
|
20 |
"W", # PyCodeStyle warnings
|
|
|
21 |
"I", # Sort imports properly
|
22 |
"UP", # Warn if certain things can changed due to newer Python versions
|
23 |
"C4", # Catch incorrect use of comprehensions, dict, list, etc
|
|
|
24 |
"ISC", # Good use of string concatenation
|
25 |
"ICN", # Use common import conventions
|
26 |
"RET", # Good return practices
|