apunkt/rowboat
1
0
Fork 0
mirror of https://github.com/rowboatlabs/rowboat.git synced 2026-04-27 17:36:25 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add temperature to swarm

Browse source
This commit is contained in:
akhisud3195 2025-02-04 21:12:10 +05:30
parent 546c7f2f6d
commit affb6c18ca
2 changed files with 6 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

8
apps/agents/src/swarm/core.py
View file

@ -39,6 +39,7 @@ class Swarm:
model_override: str,
stream: bool,
debug: bool,
temperature: float
) -> ChatCompletionMessage:
context_variables = defaultdict(str, context_variables)
instructions = (
@ -65,6 +66,7 @@ class Swarm:
"tools": funcs_and_tools or None,
"tool_choice": agent.tool_choice,
"stream": stream,
"temperature": temperature
}
if funcs_and_tools:
@ -155,7 +157,8 @@ class Swarm:
external_tools: List[str] = [],
localize_history: bool = True,
parent_has_child_history: bool = True,
tokens_used: dict = {}
tokens_used: dict = {},
temperature: float = 0.0
) -> Response:
active_agent = agent
@ -180,7 +183,8 @@ class Swarm:
context_variables=context_variables,
model_override=model_override,
stream=stream,
debug=debug
debug=debug,
temperature=temperature
)
tokens_used = update_tokens_used(provider="openai", model=model_override or active_agent.model, tokens_used=tokens_used, completion=completion)