Update agent.py

agent.py

@@ -168,42 +168,4 @@ sup = create_client(os.getenv("SUPABASE_URL"), os.getenv("SUPABASE_SERVICE_ROLE_
 vector_store = SupabaseVectorStore(
     client=sup, embedding=emb,
     table_name=os.getenv("VECTORTABLE_NAME", "documents2"),
-    query_name=os.getenv("VECTOR_QUERY_NAME", "match_documents_2")
-)
-
-# build_graph always uses OpenAI
-def build_graph():
-    llm = ChatOpenAI(
-        model=os.getenv("OPENAI_MODEL", "gpt-3.5-turbo"),
-        temperature=0,
-        openai_api_key=os.getenv("OPENAI_API_KEY")
-    )
-    llm_with_tools = llm.bind_tools(tools)
-
-    def retriever(state: MessagesState):
-        query = state["messages"][0].content
-        hits = vector_store.similarity_search(query, k=1)
-        if hits:
-            return {"messages": [sys_msg, HumanMessage(content=hits[0].page_content)]}
-        return {"messages": [sys_msg] + state["messages"]}
-
-    def assistant(state: MessagesState):
-        resp = llm_with_tools.invoke(state["messages"])
-        return {"messages": [resp]}
-
-    builder = StateGraph(MessagesState)
-    builder.add_node("retriever", retriever)
-    builder.add_node("assistant", assistant)
-    builder.add_node("tools", ToolNode(tools))
-    builder.add_edge(START, "retriever")
-    builder.add_edge("retriever", "assistant")
-    builder.add_conditional_edges("assistant", tools_condition)
-    builder.add_edge("tools", "assistant")
-    return builder.compile()
-
-# optional test
-if __name__ == "__main__":
-    graph = build_graph()
-    msgs = graph.invoke({"messages": [HumanMessage(content="Hello world")]})
-    for m in msgs["messages"]:
-        print(m.content)
+    query_name=os.getenv("VECTOR_QUERY_NAME","match_documents_langchain")