使用原生的Crewai框架
在自定义的工具llm_utils.py
或是crew.py
文件里如下配置(调用OpenAI)
from crewai import Agent, Crew, Process, Task, LLM
from crewai.project import CrewBase, agent, crew, task
llm = LLM(
base_url="https://xxxxxxx/v1",
api_key="sk-xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx",
model=O"gpt-4o", # 本次使用的模型
# temperature=0.7, # 发散的程度
# timeout=None,# 服务请求超时
# max_retries=2,# 失败重试最大次数
)
@agent
def product_manager(self) -> agent:
return Agent(
config=self.agents_config["你配置的角色名(这不是重点)"],
llm=self.llm,
verbose=True,
)
# .................
在main.py
里做如下配置
os.environ["OPENAI_API_BASE"] = "https://xxxxxxx/v1"
os.environ["OPENAI_API_KEY"] = "sk-xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx"
os.environ[