Spaces:
Build error
Build error
from langchain.memory import ConversationBufferWindowMemory | |
from langchain_community.chat_models import ChatOpenAI | |
from langchain_mongodb.chat_message_histories import MongoDBChatMessageHistory | |
# from langchain_experimental.data_anonymizer import PresidioReversibleAnonymizer | |
from langchain.agents import AgentExecutor | |
from langchain.agents.format_scratchpad import format_to_openai_functions | |
from langchain.agents.output_parsers import OpenAIFunctionsAgentOutputParser | |
from langchain.prompts import ChatPromptTemplate, MessagesPlaceholder | |
from langchain.schema.runnable import RunnablePassthrough | |
from langchain_core.utils.function_calling import convert_to_openai_function | |
from database_functions import set_chat_bot_name,isFirstSession | |
from utils import deanonymizer, create_agent | |
import time | |
is_second = False | |
use_extra = False | |
def chat_conversations(query,user_id): | |
is_first = False | |
global is_second ,use_extra | |
if is_second: | |
if 'no' in query: | |
query= """ "Okk no problem! | |
What's up today? Need ✨ Advice, ✨ a Mood Boost, ✨ a Chat, ✨ Resource Suggestions, ✨ App Features help? How can I help?" | |
Ask this question only without thinking anything. | |
""" | |
is_second = False | |
else: | |
use_extra = True | |
if query == "START": | |
if isFirstSession(user_id): | |
is_second = True | |
query = """ return this message without changing it.:- | |
also don't thought about it. Hey! I'm your BMOXI AI bestie, ready to help you tackle the wild ride of teen life. Want to give me a name? Type it below, or just say 'no' if you're cool with 'AI Bestie'!""" | |
else: | |
is_first = True | |
query = "" | |
# anonymizer = PresidioReversibleAnonymizer( | |
# analyzed_fields=["PHONE_NUMBER", | |
# "EMAIL_ADDRESS", "CREDIT_CARD"], | |
# faker_seed=42, | |
# ) | |
# anonymized_input = anonymizer.anonymize( | |
# query | |
# ) | |
start = time.time() | |
agent = create_agent(user_id,is_first) | |
print("time to create agent: ",time.time()-start) | |
response = agent({"input": query})['output'] | |
print(response) | |
print("time to generate response by agent",time.time()-start) | |
# response = chat_balancer(query,response) | |
if "Okay, from now my name will be " in response: | |
set_chat_bot_name(response.split("Okay, from now my name will be ")[-1], user_id) | |
if use_extra: | |
use_extra = False | |
is_second = False | |
return response+ "\n What's up today? Need ✨ Advice, ✨ a Mood Boost, ✨ a Chat, ✨ Resource Suggestions, ✨ App Features help? How can I help?" | |
return response | |
# output = deanonymizer(response, anonymizer) | |
return response | |