From 0a8549d4214c7b643a65745b666710c134bd81aa Mon Sep 17 00:00:00 2001
From: sinergia <alejandro.ayala@onecluster.org>
Date: Thu, 7 Nov 2024 17:17:25 -0500
Subject: [PATCH] Feat: Bash Interaction Promting

---
 app/langchain_tools/agent_tools.py | 12 ++++++------
 app/server.py                      | 20 +++++++++++++++++---
 2 files changed, 23 insertions(+), 9 deletions(-)

diff --git a/app/langchain_tools/agent_tools.py b/app/langchain_tools/agent_tools.py
index 9e513f4..bdf756f 100644
--- a/app/langchain_tools/agent_tools.py
+++ b/app/langchain_tools/agent_tools.py
@@ -6,12 +6,12 @@ from google.oauth2.credentials import Credentials
 from google_auth_oauthlib.flow import InstalledAppFlow
 from google.auth.transport.requests import Request
 from googleapiclient.discovery import build
-from ..rag.split_docs import load_split_docs
-from ..rag.llm import load_llm_openai
-from ..rag.embeddings import load_embeddins
-from ..rag.retriever import create_retriever
-from ..rag.vectorstore import create_verctorstore
-from ..rag.rag_chain import create_rag_chain
+from app.rag.split_docs import load_split_docs
+from app.rag.llm import load_llm_openai
+from app.rag.embeddings import load_embeddins
+from app.rag.retriever import create_retriever
+from app.rag.vectorstore import create_verctorstore
+from app.rag.rag_chain import create_rag_chain
 import pytz
 import telebot
 import os
diff --git a/app/server.py b/app/server.py
index 3e1a5ac..af588a9 100644
--- a/app/server.py
+++ b/app/server.py
@@ -11,7 +11,7 @@ from langchain_community.tools.gmail.utils import (
     get_gmail_credentials)
 from langchain_community.agent_toolkits import GmailToolkit
 
-from .langchain_tools.agent_tools import (
+from app.langchain_tools.agent_tools import (
     redact_email,
     list_calendar_events,
     create_calendar_event,
@@ -155,6 +155,20 @@ async def process_text(request: Request):
 
 
 if __name__ == "__main__":
-    import uvicorn
+    config = {"configurable": {"thread_id": "thread-1", "recursion_limit": 50}}
 
-    uvicorn.run(app, host="0.0.0.0", port=8000)
+    while True:
+        user_input = input("User: ")
+        if user_input.lower() in ["quit", "exit", "q"]:
+            print("Goodbye!")
+            break
+
+        events = graph.stream({
+            "messages": [("user", user_input)],
+            "is_last_step": False},
+                              config, stream_mode="updates")
+
+        for event in events:
+            if "agent" in event:
+                print(
+                    f"\nAsistente: {event['agent']['messages'][-1].content}\n")