pygeai 0.4.0b4__tar.gz → 0.4.0b5__tar.gz

{pygeai-0.4.0b4/pygeai.egg-info → pygeai-0.4.0b5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pygeai
-Version: 0.4.0b4
+Version: 0.4.0b5
 Summary: Software Development Kit to interact with Globant Enterprise AI.
 Author-email: Globant <geai-sdk@globant.com>
 License-Expression: MIT

{pygeai-0.4.0b4 → pygeai-0.4.0b5}/pygeai/cli/commands/chat.py

@@ -13,12 +13,14 @@ from pygeai.chat.session import AgentChatSession
 from pygeai.cli.commands import Command, Option, ArgumentsEnum
 from pygeai.cli.commands.builders import build_help_text
 from pygeai.cli.commands.common import get_messages, get_boolean_value, get_penalty_float_value
+from pygeai.cli.commands.lab.utils import get_project_id
 from pygeai.cli.texts.help import CHAT_HELP_TEXT
-from pygeai.core.common.exceptions import MissingRequirementException, WrongArgumentError
+from pygeai.core.common.exceptions import MissingRequirementException, WrongArgumentError, InvalidAgentException
 from prompt_toolkit import PromptSession
 from prompt_toolkit.history import InMemoryHistory
 
 from pygeai.core.utils.console import Console
+from pygeai.lab.agents.clients import AgentClient
 
 
 def show_help():
@@ -359,9 +361,13 @@ def chat_with_agent(option_list: list):
     if not agent_name:
         raise MissingRequirementException(f"Agent name must be specified.")
 
+    project_id = get_project_id()
+    agent_data = AgentClient().get_agent(project_id=project_id, agent_id=agent_name)
+    if 'errors' in agent_data:
+        raise InvalidAgentException(f"There is no agent with that name: {agent_data.get('errors')}")
+
     if use_gui:
         try:
-            # Check if Streamlit is installed
             import streamlit
         except ImportError:
             logger.error("Streamlit not installed")