Merge pull request #176 from askui/feat/configure-temperature

feat/configure temperature

Author: adi-wan-askui

src/askui/models/anthropic/messages_api.py

@@ -64,6 +64,7 @@ def create_message(
         system: str | list[BetaTextBlockParam] | Omit = omit,
         thinking: BetaThinkingConfigParam | Omit = omit,
         tool_choice: BetaToolChoiceParam | Omit = omit,
+        temperature: float | Omit = omit,
     ) -> MessageParam:
         _messages = [
             cast("BetaMessageParam", message.model_dump(exclude={"stop_reason"}))
@@ -78,6 +79,7 @@ def create_message(
             system=system,
             thinking=thinking,
             tool_choice=tool_choice,
+            temperature=temperature,
             timeout=300.0,
         )
         return MessageParam.model_validate(response.model_dump())

src/askui/models/shared/agent.py

@@ -86,6 +86,7 @@ def _step(
                 system=settings.messages.system,
                 thinking=settings.messages.thinking,
                 tool_choice=settings.messages.tool_choice,
+                temperature=settings.messages.temperature,
             )
             message_by_assistant = self._call_on_message(
                 on_message, response_message, truncation_strategy.messages

src/askui/models/shared/messages_api.py

@@ -26,6 +26,7 @@ def create_message(
         system: str | list[BetaTextBlockParam] | Omit = omit,
         thinking: BetaThinkingConfigParam | Omit = omit,
         tool_choice: BetaToolChoiceParam | Omit = omit,
+        temperature: float | Omit = omit,
     ) -> MessageParam:
         """Create a message using the Anthropic API.
 
@@ -38,6 +39,7 @@ def create_message(
             system (str | list[BetaTextBlockParam] | Omit): The system to use.
             thinking (BetaThinkingConfigParam | Omit): The thinking to use.
             tool_choice (BetaToolChoiceParam | Omit): The tool choice to use.
+            temperature (float | Omit): The temperature to use.
 
         Returns:
             MessageParam: The created message.

src/askui/models/shared/settings.py

@@ -18,6 +18,7 @@ class MessageSettings(BaseModel):
     system: str | list[BetaTextBlockParam] | Omit = omit
     thinking: BetaThinkingConfigParam | Omit = omit
     tool_choice: BetaToolChoiceParam | Omit = omit
+    temperature: float | Omit = Field(default=omit, ge=0.0, le=1.0)
 
 
 class ActSettings(BaseModel):