pygeai 0.3.2__py3-none-any.whl → 0.4.0__py3-none-any.whl

pygeai/__init__.py

@@ -4,7 +4,7 @@ import sys
 from pathlib import Path
 
 __author__ = 'Globant'
-__version__ = '0.3.2'
+__version__ = '0.4.0'
 
 # Add vendor directory to Python path
 package_root = Path(__file__).parent

pygeai/chat/ui.py

@@ -1,5 +1,4 @@
 import logging
-from pathlib import Path
 
 import streamlit as st
 import argparse

pygeai/cli/__init__.py

@@ -4,5 +4,5 @@ GEAI - ClI
 Command line interface to interact with Globant Enterprise AI.
 """
 
-__version__ = '0.3.2'
+__version__ = '0.4.0'
 

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,71 +361,67 @@ 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(project_id=project_id).get_agent(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")
             Console.write_stderr("Streamlit is required for GUI mode. Install it with 'pip install streamlit'.")
             sys.exit(1)
 
-        if use_gui:
-            try:
-                import streamlit
-            except ImportError:
-                logger.error("Streamlit not installed")
-                Console.write_stderr("Streamlit is required for GUI mode. Install it with 'pip install streamlit'.")
-                sys.exit(1)
+        try:
+            ui_path = resources.files("pygeai.chat").joinpath("ui.py")
+            ui_file_path = str(ui_path)
+
+            # Add the top-level project root to PYTHONPATH
+            package_root = str(Path(ui_file_path).resolve().parents[2])
+            env = os.environ.copy()
+            env["PYTHONPATH"] = package_root + os.pathsep + env.get("PYTHONPATH", "")
+
+            streamlit_cmd = [
+                sys.executable, "-m", "streamlit", "run", ui_file_path,
+                "--server.address", "127.0.0.1",
+                "--", "--agent-name", agent_name
+            ]
+
+            if platform.system() == "Linux":
+                streamlit_cmd.insert(5, "--server.headless=true")
+
+            process = subprocess.Popen(
+                streamlit_cmd,
+                stdout=subprocess.PIPE,
+                stderr=subprocess.PIPE,
+                text=True
+            )
 
-            try:
-                ui_path = resources.files("pygeai.chat").joinpath("ui.py")
-                ui_file_path = str(ui_path)
-
-                # Add the top-level project root to PYTHONPATH
-                package_root = str(Path(ui_file_path).resolve().parents[2])
-                env = os.environ.copy()
-                env["PYTHONPATH"] = package_root + os.pathsep + env.get("PYTHONPATH", "")
-
-                streamlit_cmd = [
-                    sys.executable, "-m", "streamlit", "run", ui_file_path,
-                    "--server.address", "127.0.0.1",
-                    "--", "--agent-name", agent_name
-                ]
-
-                if platform.system() == "Linux":
-                    streamlit_cmd.insert(5, "--server.headless=true")
-
-                process = subprocess.Popen(
-                    streamlit_cmd,
-                    stdout=subprocess.PIPE,
-                    stderr=subprocess.PIPE,
-                    text=True
-                )
-
-                url = "http://localhost:8501"
-                if platform.system() == "Linux":
-                    Console.write_stdout(f"Open Streamlit app at {url} (or next port like 8502 if 8501 is taken)")
-
-                try:
-                    stdout, stderr = process.communicate()
-                    if stderr:
-                        logger.error(f"Streamlit stderr:\n{stderr}")
-                        Console.write_stderr(f"Streamlit error:\n{stderr}")
-                except KeyboardInterrupt:
-                    process.terminate()
-                    Console.write_stdout("Streamlit stopped.")
-                    sys.exit(0)
+            url = "http://localhost:8501"
+            if platform.system() == "Linux":
+                Console.write_stdout(f"Open Streamlit app at {url} (or next port like 8502 if 8501 is taken)")
 
-            except FileNotFoundError:
-                logger.error("Could not locate pygeai/chat/ui.py")
-                Console.write_stderr("Streamlit UI file not found. Ensure pygeai is installed correctly.")
-                sys.exit(1)
-            except Exception as e:
-                logger.error(f"Streamlit error: {e}")
-                Console.write_stderr(
-                    f"Failed to launch Streamlit. Check port with 'lsof -i :8501' and kill any process, or try {url}.")
-                sys.exit(1)
+            try:
+                stdout, stderr = process.communicate()
+                if stderr:
+                    logger.error(f"Streamlit stderr:\n{stderr}")
+                    Console.write_stderr(f"Streamlit error:\n{stderr}")
+            except KeyboardInterrupt:
+                process.terminate()
+                Console.write_stdout("Streamlit stopped.")
+                sys.exit(0)
+
+        except FileNotFoundError:
+            logger.error("Could not locate pygeai/chat/ui.py")
+            Console.write_stderr("Streamlit UI file not found. Ensure pygeai is installed correctly.")
+            sys.exit(1)
+        except Exception as e:
+            logger.error(f"Streamlit error: {e}")
+            Console.write_stderr(
+                f"Failed to launch Streamlit. Check port with 'lsof -i :8501' and kill any process, or try {url}.")
+            sys.exit(1)
     else:
         chat_session = AgentChatSession(agent_name)
         messages = list()