You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
36 lines
868 B
36 lines
868 B
import os
|
|
|
|
from swarms.models import OpenAIChat
|
|
from swarms.models.bing_chat import BingChat
|
|
from swarms.tools.autogpt import EdgeGPTTool, tool
|
|
from swarms.workers.worker import Worker
|
|
|
|
api_key = os.getenv("OPENAI_API_KEY")
|
|
|
|
# Initialize the EdgeGPTModel
|
|
edgegpt = BingChat(cookies_path="./cookies.txt")
|
|
|
|
|
|
@tool
|
|
def edgegpt(task: str = None):
|
|
"""A tool to run infrence on the EdgeGPT Model"""
|
|
return EdgeGPTTool.run(task)
|
|
|
|
|
|
# Initialize the language model,
|
|
# This model can be swapped out with Anthropic, ETC, Huggingface Models like Mistral, ETC
|
|
llm = OpenAIChat(
|
|
openai_api_key=api_key,
|
|
temperature=0.5,
|
|
)
|
|
|
|
# Initialize the Worker with the custom tool
|
|
worker = Worker(
|
|
llm=llm, ai_name="EdgeGPT Worker", external_tools=[edgegpt]
|
|
)
|
|
|
|
# Use the worker to process a task
|
|
task = "Hello, my name is ChatGPT"
|
|
response = worker.run(task)
|
|
print(response)
|