sampsong commited on
Commit
32b3ca4
·
1 Parent(s): accc373

add prompt

Browse files
Files changed (2) hide show
  1. Agents/agent.py +3 -1
  2. app.py +12 -2
Agents/agent.py CHANGED
@@ -242,10 +242,12 @@ def build_graph(provider: str="groq"):
242
  llmWithTools = llm.bind_tools(tools)
243
 
244
  def assistant(state: MessagesState):
 
245
  msg = {"messages": [sysMsg] + state["messages"]}
 
246
  #print(f"invoke llmwithtools with state message = {msg}")
247
  #msgbind = llmWithTools.invoke(state["messages"])
248
- #print(f"msgBind{msgbind}")
249
  return {"messages": [llmWithTools.invoke(msg)]}
250
  '''
251
  def retriever(state: MessagesState):
 
242
  llmWithTools = llm.bind_tools(tools)
243
 
244
  def assistant(state: MessagesState):
245
+ #msg
246
  msg = {"messages": [sysMsg] + state["messages"]}
247
+
248
  #print(f"invoke llmwithtools with state message = {msg}")
249
  #msgbind = llmWithTools.invoke(state["messages"])
250
+ print(f"msg: {msg}")
251
  return {"messages": [llmWithTools.invoke(msg)]}
252
  '''
253
  def retriever(state: MessagesState):
app.py CHANGED
@@ -3,7 +3,10 @@ import gradio as gr
3
  import requests
4
  import inspect
5
  import pandas as pd
6
- from langchain_core.messages import HumanMessage
 
 
 
7
  from Agents.agent import build_graph, tools,systemPrompt
8
  from langfuse.langchain import CallbackHandler
9
  from langgraph.prebuilt import create_react_agent
@@ -42,12 +45,19 @@ class BasicAgent:
42
  self.graph = build_graph()
43
  def __call__(self, question: str, imageURL: str="") -> str:
44
  print(f"Agent received question (first 100 chars): {question[:100]}...")
45
- if(imageURL.strip == ""):
46
  messages = [HumanMessage(content=question)]
47
  else:
48
  formattedImageURL = gaiaValidationURL + imageURL
49
  messages = [HumanMessage(content=question, additional_kwargs={"imageURL":formattedImageURL })]
50
 
 
 
 
 
 
 
 
51
  if(not langFuseOn):
52
  print("no langfuse")
53
  messages = self.graph.invoke({"messages": messages})
 
3
  import requests
4
  import inspect
5
  import pandas as pd
6
+ from langchain_core.output_parsers import JsonOutputParser
7
+ from langchain_core.prompts import ChatMessagePromptTemplate, ChatPromptTemplate
8
+ from langchain_core.tools import render_text_description
9
+ from langchain_core.messages import HumanMessage, SystemMessage
10
  from Agents.agent import build_graph, tools,systemPrompt
11
  from langfuse.langchain import CallbackHandler
12
  from langgraph.prebuilt import create_react_agent
 
45
  self.graph = build_graph()
46
  def __call__(self, question: str, imageURL: str="") -> str:
47
  print(f"Agent received question (first 100 chars): {question[:100]}...")
48
+ if(imageURL.len == 0):
49
  messages = [HumanMessage(content=question)]
50
  else:
51
  formattedImageURL = gaiaValidationURL + imageURL
52
  messages = [HumanMessage(content=question, additional_kwargs={"imageURL":formattedImageURL })]
53
 
54
+ '''prompt = ChatPromptTemplate.from_messages([
55
+ SystemMessage(content="systemPrompt"),
56
+ ("human", "{question}"),
57
+ ])
58
+ '''
59
+
60
+
61
  if(not langFuseOn):
62
  print("no langfuse")
63
  messages = self.graph.invoke({"messages": messages})