jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

Add temperature parameter to ConversationalAgent

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-12-20 21:06:09 -03:00
commit cfcfc37928
1 changed files with 10 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

10
src/backend/langflow/components/agents/OpenAIConversationalAgent.py
View file

@ -9,7 +9,9 @@ from langchain.prompts import SystemMessagePromptTemplate
from langchain.prompts.chat import MessagesPlaceholder from langchain.prompts.chat import MessagesPlaceholder
from langchain.schema.memory import BaseMemory from langchain.schema.memory import BaseMemory
from langchain.tools import Tool from langchain.tools import Tool
from langflow import CustomComponent from langflow import CustomComponent
from langflow.field_typing.range_spec import RangeSpec
class ConversationalAgent(CustomComponent): class ConversationalAgent(CustomComponent):
@ -35,6 +37,11 @@ class ConversationalAgent(CustomComponent):
"value": openai_function_models[0], "value": openai_function_models[0],
}, },
"code": {"show": False}, "code": {"show": False},
"temperature": {
"display_name": "Temperature",
"value": 0.2,
"range_spec": RangeSpec(min=0, max=2, step=0.1),
},
} }
def build( def build(
@ -46,11 +53,14 @@ class ConversationalAgent(CustomComponent):
memory: Optional[BaseMemory] = None, memory: Optional[BaseMemory] = None,
system_message: Optional[SystemMessagePromptTemplate] = None, system_message: Optional[SystemMessagePromptTemplate] = None,
max_token_limit: int = 2000, max_token_limit: int = 2000,
temperature: float = 0.9,
) -> AgentExecutor: ) -> AgentExecutor:
llm = ChatOpenAI( llm = ChatOpenAI(
model=model_name, model=model_name,
api_key=openai_api_key, api_key=openai_api_key,
base_url=openai_api_base, base_url=openai_api_base,
max_tokens=max_token_limit,
temperature=temperature,
) )
if not memory: if not memory:
memory_key = "chat_history" memory_key = "chat_history"