mirror of
https://github.com/IntelligenzaArtificiale/Free-Auto-GPT.git
synced 2026-07-03 13:15:41 +02:00
IntelligenzaArtificiale
187 lines
7.7 KiB
Python
187 lines
7.7 KiB
Python
from langchain.prompts.chat import (
|
|
SystemMessagePromptTemplate,
|
|
HumanMessagePromptTemplate,
|
|
)
|
|
from langchain.schema import (
|
|
AIMessage,
|
|
HumanMessage,
|
|
SystemMessage,
|
|
BaseMessage,
|
|
)
|
|
|
|
from langchain.llms.base import LLM
|
|
from typing import Optional, List, Mapping, Any
|
|
#from t3nsorAPI import gpt3NoInternet not working the best
|
|
#from quoraAPI import GPT4QUORA not working
|
|
from sqlchatAPI import sqlchatGPT3 #work but low result
|
|
from phindAPI import phindGPT4Internet
|
|
from writesonicAPI import writesonicGPT3Internet
|
|
from youAPI import youGPT3Internet
|
|
from message import get_sys_msgs
|
|
import streamlit as st
|
|
from streamlit_chat_media import message
|
|
|
|
st.set_page_config(
|
|
page_title="FREE AUTOGPT 🚀 by Intelligenza Artificiale Italia",
|
|
page_icon="🚀",
|
|
layout="wide",
|
|
menu_items={
|
|
'Get help': 'https://www.intelligenzaartificialeitalia.net/',
|
|
'Report a bug': "mailto:servizi@intelligenzaartificialeitalia.net",
|
|
'About': "# *🚀 FREE AUTOGPT 🚀* "
|
|
}
|
|
)
|
|
|
|
|
|
st.markdown("<style> iframe > div { text-align: left;} </style>", unsafe_allow_html=True)
|
|
|
|
class CAMELAgent:
|
|
|
|
def __init__(
|
|
self,
|
|
system_message: SystemMessage,
|
|
model: writesonicGPT3Internet,
|
|
) -> None:
|
|
self.system_message = system_message.content
|
|
self.model = model
|
|
self.init_messages()
|
|
|
|
def reset(self) -> None:
|
|
self.init_messages()
|
|
return self.stored_messages
|
|
|
|
def init_messages(self) -> None:
|
|
self.stored_messages = [self.system_message]
|
|
|
|
def update_messages(self, message: BaseMessage) -> List[BaseMessage]:
|
|
self.stored_messages.append(message)
|
|
return self.stored_messages
|
|
|
|
def step(
|
|
self,
|
|
input_message: HumanMessage,
|
|
) -> AIMessage:
|
|
messages = self.update_messages(input_message)
|
|
output_message = self.model(str(input_message.content))
|
|
self.update_messages(output_message)
|
|
print(f"AI Assistant:\n\n{output_message}\n\n")
|
|
return output_message
|
|
|
|
|
|
|
|
|
|
col1, col2 = st.columns(2)
|
|
assistant_role_name = col1.text_input("Assistant Role Name", "Python Programmer")
|
|
user_role_name = col2.text_input("User Role Name", "Stock Trader")
|
|
task = st.text_area("Task", "Develop a trading bot for the stock market")
|
|
word_limit = st.number_input("Word Limit", 10, 1500, 50)
|
|
|
|
#choose a model for agent
|
|
cola, colb = st.columns(2)
|
|
assistant_model = cola.selectbox("Assistant Model", ["PHIND + Internet (GPT4)", "YOU + Internet (GPT3.5)", "WRITESONIC + Internet (GPT3.5)", "SQLCHAT.AI (GPT3.5 turbo)"])
|
|
user_model = colb.selectbox("User Model", ["PHIND + Internet (GPT4)", "YOU + Internet (GPT3.5)", "WRITESONIC + Internet (GPT3.5)", "SQLCHAT.AI (GPT3.5 turbo)"])
|
|
|
|
if assistant_model == "YOU + Internet (GPT3.5)":
|
|
assistant_model = youGPT3Internet()
|
|
elif assistant_model == "WRITESONIC + Internet (GPT3.5)":
|
|
assistant_model = writesonicGPT3Internet()
|
|
elif assistant_model == "SQLCHAT.AI (GPT3.5 turbo)":
|
|
assistant_model = sqlchatGPT3()
|
|
|
|
if user_model == "YOU + Internet (GPT3.5)":
|
|
user_model = youGPT3Internet()
|
|
elif user_model == "WRITESONIC + Internet (GPT3.5)":
|
|
user_model = writesonicGPT3Internet()
|
|
elif user_model == "SQLCHAT.AI (GPT3.5 turbo)":
|
|
user_model = sqlchatGPT3()
|
|
|
|
if st.button("Start Autonomus AI AGENT"):
|
|
task_specifier_sys_msg = SystemMessage(content="You can make a task more specific.")
|
|
task_specifier_prompt = (
|
|
"""Here is a task that {assistant_role_name} will help {user_role_name} to complete: {task}.
|
|
Please make it more specific. Be creative and imaginative.
|
|
Please reply with the specified task in {word_limit} words or less. Do not add anything else."""
|
|
)
|
|
task_specifier_template = HumanMessagePromptTemplate.from_template(template=task_specifier_prompt)
|
|
|
|
task_specify_agent = CAMELAgent(task_specifier_sys_msg, writesonicGPT3Internet())
|
|
task_specifier_msg = task_specifier_template.format_messages(assistant_role_name=assistant_role_name,
|
|
user_role_name=user_role_name,
|
|
task=task, word_limit=word_limit)[0]
|
|
|
|
specified_task_msg = task_specify_agent.step(task_specifier_msg)
|
|
|
|
|
|
print(f"Specified task: {specified_task_msg}")
|
|
message(f"Specified task: {specified_task_msg}", allow_html=True, key="specified_task" , avatar_style="adventurer")
|
|
|
|
specified_task = specified_task_msg
|
|
|
|
#define the role system messages
|
|
assistant_sys_msg, user_sys_msg = get_sys_msgs(assistant_role_name, user_role_name, specified_task)
|
|
|
|
|
|
#define the agents
|
|
|
|
#for GPT4 + internet : phindGPT4Internet(base_prompt=assistant_sys_msg.content) or phindGPT4Internet(messages=[], base_prompt=user_sys_msg.content)
|
|
|
|
#for YOU + internt : youGPT3Internet()
|
|
|
|
#for WRITESONIC + internt : writesonicGPT3Internet()
|
|
|
|
#for sqlchat.ai + internt : sqlchatGPT3()
|
|
|
|
#for t3nsor gpt3NoInternet() - NOT WORKING BEST RESULT
|
|
|
|
#for Quora GPT4QUORA() - WORKING 50%
|
|
|
|
if assistant_model == "PHIND + Internet (GPT4)":
|
|
assistant_model = phindGPT4Internet(base_prompt=assistant_sys_msg.content)
|
|
|
|
if user_model == "PHIND + Internet (GPT4)":
|
|
user_model = phindGPT4Internet(base_prompt=user_sys_msg.content)
|
|
|
|
|
|
message(f"Starting AI ASSISTANT '{assistant_role_name}' with assigned model: {assistant_model}", allow_html=True, key="assistant_model" , avatar_style="pixel-art-neutral")
|
|
message(f"Starting AI USER '{user_role_name}' with assigned model: {user_model}", allow_html=True, key="user_model" , avatar_style="fun-emoji")
|
|
|
|
#AI ASSISTANT setup |-> add the agent LLM MODEL HERE <-|
|
|
assistant_agent = CAMELAgent(assistant_sys_msg, phindGPT4Internet(messages=[],base_prompt=assistant_sys_msg.content))
|
|
|
|
#AI USER setup |-> add the agent LLM MODEL HERE <-|
|
|
user_agent = CAMELAgent(user_sys_msg, phindGPT4Internet(messages=[], base_prompt=user_sys_msg.content))
|
|
|
|
# Reset agents
|
|
assistant_agent.reset()
|
|
user_agent.reset()
|
|
|
|
# Initialize chats
|
|
assistant_msg = HumanMessage(
|
|
content=(f"{user_sys_msg}. "
|
|
"Now start to give me introductions one by one. "
|
|
"Only reply with Instruction and Input."))
|
|
|
|
user_msg = HumanMessage(content=f"{assistant_sys_msg.content}")
|
|
user_msg = assistant_agent.step(user_msg)
|
|
message(f"AI Assistant ({assistant_role_name}):\n\n{user_msg}\n\n", is_user=False, allow_html=True, key="0_assistant" , avatar_style="pixel-art")
|
|
print(f"Original task prompt:\n{task}\n")
|
|
print(f"Specified task prompt:\n{specified_task}\n")
|
|
|
|
chat_turn_limit, n = 30, 0
|
|
while n < chat_turn_limit:
|
|
n += 1
|
|
user_ai_msg = user_agent.step(assistant_msg)
|
|
user_msg = HumanMessage(content=user_ai_msg)
|
|
#print(f"AI User ({user_role_name}):\n\n{user_msg}\n\n")
|
|
message(f"AI User ({user_role_name}):\n\n{user_msg.content}\n\n", is_user=True, allow_html=True, key=str(n)+"_user")
|
|
|
|
assistant_ai_msg = assistant_agent.step(user_msg)
|
|
assistant_msg = HumanMessage(content=assistant_ai_msg)
|
|
#print(f"AI Assistant ({assistant_role_name}):\n\n{assistant_msg}\n\n")
|
|
message(f"AI Assistant ({assistant_role_name}):\n\n{assistant_msg.content}\n\n", is_user=False, allow_html=True, key=str(n)+"_assistant" , avatar_style="pixel-art")
|
|
if "<CAMEL_TASK_DONE>" in user_msg.content or 'task completed' in user_msg.content:
|
|
message("Task completed!", allow_html=True, key="task_done")
|
|
break
|
|
if 'Error' in user_msg.content:
|
|
message("Task failed!", allow_html=True, key="task_failed")
|
|
break |