File size: 794 Bytes
16d5a75
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
from langgraph.graph import StateGraph, START, END
from src.config.llm import llm_2_0
from .func import State
from langgraph.graph.state import CompiledStateGraph
from src.agents.prompt_analyzed.flow import analyze_prompt, analyze_agent
from src.agents.react_agent.prompt import ReAgent_prompt
from langgraph.prebuilt import create_react_agent
from typing import TypedDict, List

def enough_information(name,info):
    """
    gọi tool này sau khi hoàn tất quá trình thu thập thông tin cần thiết
    Args:
        name (str): Tên chatbot cần tạo
        info (str): Thông tin đã thu thập được
    """
    print(info)
    return "Created successful"

react_agent = create_react_agent(
    model=llm_2_0,
    tools=[enough_information],
    prompt=ReAgent_prompt,

)