En mi caso tuve muchos problemas con las importaciones mostradas en la clase, asi que estuve investigando y probando hasta que en mi caso funciono con la siguiente estructura.
Espero que les sea de mucha ayuda y les ahorre tiempo.
from langchain_google_genai import ChatGoogleGenerativeAI
from my_models import GEMINI_FLASH
from my_keys import GEMINI_API_KEY
from langchain_core.globals import set_debug
from langsmith import Client
hub = Client()
from langchain_core.tools import Tool
from langchain_classic.agents import create_react_agent
from herramienta_analisis_imagen import HerramientaAnalisisImagen
set_debug(False)
class AgenteOrquestador:
def __init__(self):
self.llm = ChatGoogleGenerativeAI(
api_key=GEMINI_API_KEY,
model=GEMINI_FLASH
)
herramienta_analisis_imagen = HerramientaAnalisisImagen()
self.tools =[
Tool(
name = herramienta_analisis_imagen.name,
func = herramienta_analisis_imagen.run,
description = herramienta_analisis_imagen.description,
return_direct = herramienta_analisis_imagen.return_direct
)
]
prompt = hub.pull_prompt("hwchase17/react", dangerously_pull_public_prompt=True)
self.agente = create_react_agent(self.llm,self.tools,prompt)
