wehub-resource-sync
61 lines
1.5 KiB
Python
61 lines
1.5 KiB
Python
import os
|
|
|
|
import dotenv
|
|
from autogen import AssistantAgent, UserProxyAgent
|
|
from composio_autogen import AutogenProvider
|
|
|
|
from composio import Composio
|
|
|
|
# Load environment variables from .env
|
|
dotenv.load_dotenv()
|
|
|
|
|
|
def is_termination_msg(content: dict) -> bool:
|
|
"""Check if a message contains termination message."""
|
|
return "TERMINATE" in (content.get("content", "") or "")
|
|
|
|
|
|
def main():
|
|
# Initialize tools.
|
|
chatbot = AssistantAgent(
|
|
"chatbot",
|
|
system_message="Reply TERMINATE when the task is done or when user's content is empty",
|
|
llm_config={
|
|
"config_list": [
|
|
{"model": "gpt-5", "api_key": os.environ["OPENAI_API_KEY"]},
|
|
]
|
|
},
|
|
)
|
|
|
|
# Create a user proxy agent
|
|
user_proxy = UserProxyAgent(
|
|
"user_proxy",
|
|
is_termination_msg=is_termination_msg,
|
|
human_input_mode="NEVER",
|
|
code_execution_config={"use_docker": False},
|
|
)
|
|
|
|
# Get composio tools
|
|
composio = Composio(provider=AutogenProvider())
|
|
tools = composio.tools.get(user_id="default", toolkits=["GITHUB"])
|
|
|
|
# Register the preferred Applications, with right executor.
|
|
composio.provider.register_tools(
|
|
caller=chatbot,
|
|
executor=user_proxy,
|
|
tools=tools,
|
|
)
|
|
|
|
# Define task.
|
|
task = "Star a repo composiohq/composio on GitHub"
|
|
|
|
# Execute task.
|
|
response = user_proxy.initiate_chat(chatbot, message=task)
|
|
|
|
# Print response
|
|
print(response.chat_history)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|