andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-01-10 00:28:31 +00:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Updating openai version

Browse Source
This commit is contained in:
João Moura
2023-12-20 17:21:48 -03:00
parent e0270d6a68
commit 2bd15411da
3 changed files with 93 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
crewai/agent.py
View File

@@ -4,7 +4,7 @@ from typing import List, Any, Optional
from pydantic.v1 import BaseModel, PrivateAttr, Field, root_validator
from langchain.agents import AgentExecutor
from langchain.chat_models import ChatOpenAI as OpenAI
from langchain.chat_models import ChatOpenAI
from langchain.tools.render import render_text_description
from langchain.agents.format_scratchpad import format_log_to_str
from langchain.agents.output_parsers import ReActSingleInputOutputParser
@@ -40,7 +40,7 @@ class Agent(BaseModel):
@root_validator(pre=True)
def check_llm(_cls, values):
if not values.get('llm'):
values['llm'] = OpenAI(
values['llm'] = ChatOpenAI(
temperature=0.7,
model_name="gpt-4"
)
@@ -60,8 +60,6 @@ class Agent(BaseModel):
"handle_parsing_errors": True,
}
llm_with_bind = self.llm.bind(stop=["\nObservation"])
if self.memory:
summary_memory = ConversationSummaryMemory(
llm=self.llm,
@@ -79,7 +77,9 @@ class Agent(BaseModel):
role=self.role,
backstory=self.backstory,
)
inner_agent = agent_args | execution_prompt | llm_with_bind | ReActSingleInputOutputParser()
bind = self.llm.bind(stop=["\nObservation"])
inner_agent = agent_args | execution_prompt | bind | ReActSingleInputOutputParser()
self.agent_executor = AgentExecutor(
agent=inner_agent,