Commit 97829443 authored by Natalia Grigoriadou's avatar Natalia Grigoriadou
Browse files

fixed pipeline

parent fedd2b54
Loading
Loading
Loading
Loading
+2 −1
Changes for ai_agent/sub_agents/groq_agent.py: 2 added lines, 1 removed line.
Original line number Diff line number Diff line
@@ -4,6 +4,7 @@ import os
from dotenv import load_dotenv
from langchain_groq import ChatGroq
from mcp_use import MCPAgent, MCPClient  # type: ignore[import-untyped]
from pydantic import SecretStr

from ai_agent.prompts import CONVERSATION_MEMORY_PROMPT

@@ -57,7 +58,7 @@ async def create_groq_agent(authorization: str):
        reasoning_format="parsed",
        timeout=15,
        max_retries=3,
        api_key=groq_api_key,
        api_key=SecretStr(groq_api_key),
    )

    agent = MCPAgent(