Spaces:
Sleeping
Sleeping
Update agent.py
Browse files
agent.py
CHANGED
|
@@ -90,8 +90,7 @@ system_prompt = """You are a general AI assistant. I will ask you a question. Us
|
|
| 90 |
FINAL ANSWER: [YOUR FINAL ANSWER]. YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated list of numbers and/or strings.
|
| 91 |
If you are asked for a number, don't use comma to write your number neither use units such as $ or percent sign unless specified otherwise.
|
| 92 |
If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities), and write the digits in plain text unless specified otherwise.
|
| 93 |
-
If you are asked for a comma separated list, apply the above rules depending of whether the element to be put in the list is a number or a string.
|
| 94 |
-
Remember: Quality over speed. Take time to find accurate information rather than guessing."""
|
| 95 |
#Using your tools to 추가하니 툴컬링 하게됨
|
| 96 |
system_message = SystemMessage(content=system_prompt)
|
| 97 |
|
|
@@ -101,7 +100,7 @@ tools = [
|
|
| 101 |
|
| 102 |
|
| 103 |
def build_graph(provider: str = "google"):
|
| 104 |
-
llm = ChatGoogleGenerativeAI(model="gemini-2.
|
| 105 |
|
| 106 |
llm_with_tools = llm.bind_tools(tools)
|
| 107 |
|
|
|
|
| 90 |
FINAL ANSWER: [YOUR FINAL ANSWER]. YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated list of numbers and/or strings.
|
| 91 |
If you are asked for a number, don't use comma to write your number neither use units such as $ or percent sign unless specified otherwise.
|
| 92 |
If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities), and write the digits in plain text unless specified otherwise.
|
| 93 |
+
If you are asked for a comma separated list, apply the above rules depending of whether the element to be put in the list is a number or a string."""
|
|
|
|
| 94 |
#Using your tools to 추가하니 툴컬링 하게됨
|
| 95 |
system_message = SystemMessage(content=system_prompt)
|
| 96 |
|
|
|
|
| 100 |
|
| 101 |
|
| 102 |
def build_graph(provider: str = "google"):
|
| 103 |
+
llm = ChatGoogleGenerativeAI(model="gemini-2.5-flash", temperature=0, api_key=google_api_key)
|
| 104 |
|
| 105 |
llm_with_tools = llm.bind_tools(tools)
|
| 106 |
|