change max attempts to 2 from 10

This commit is contained in:
Jason Schleifer
2025-01-03 09:42:05 +13:00
parent 22987a2a00
commit 7a50f8ebdf
11 changed files with 11 additions and 11 deletions
@@ -71,7 +71,7 @@ class gtUIAmazonBedrockStructureConfig(gtUIBaseConfig):
prompt_model = kwargs.get("prompt_model", amazonBedrockPromptModels[0])
temperature = kwargs.get("temperature", 0.7)
max_attempts = kwargs.get("max_attempts_on_fail", 10)
max_attempts = kwargs.get("max_attempts_on_fail", 2)
use_native_tools = kwargs.get("use_native_tools", False)
max_tokens = kwargs.get("max_tokens", None)
params["model"] = prompt_model
@@ -54,7 +54,7 @@ class gtUIAnthropicStructureConfig(gtUIBaseConfig):
params["model"] = kwargs.get("prompt_model", anthropicPromptModels[0])
params["temperature"] = kwargs.get("temperature", 0.7)
params["max_attempts"] = kwargs.get("max_attempts_on_fail", 10)
params["max_attempts"] = kwargs.get("max_attempts_on_fail", 2)
params["use_native_tools"] = kwargs.get("use_native_tools", True)
params["api_key"] = self.getenv(kwargs.get("api_key_env_var", DEFAULT_API_KEY))
@@ -58,7 +58,7 @@ class gtUIAzureOpenAiStructureConfig(gtUIBaseConfig):
params["model"] = kwargs.get("prompt_model", "gpt-4o")
params["temperature"] = kwargs.get("temperature", 0.7)
params["seed"] = kwargs.get("seed", 12341)
params["max_attempts"] = kwargs.get("max_attempts_on_fail", 10)
params["max_attempts"] = kwargs.get("max_attempts_on_fail", 2)
params["azure_deployment"] = kwargs.get("prompt_model_deployment_name", "gpt4o")
params["use_native_tools"] = kwargs.get("use_native_tools", False)
params["stream"] = kwargs.get("stream", False)
@@ -54,7 +54,7 @@ class gtUIGoogleStructureConfig(gtUIBaseConfig):
params = {}
temperature = kwargs.get("temperature", 0.7)
prompt_model = kwargs.get("prompt_model", google_models[0])
max_attempts = kwargs.get("max_attempts_on_fail", 10)
max_attempts = kwargs.get("max_attempts_on_fail", 2)
api_key = self.getenv(kwargs.get("api_key_env_var", DEFAULT_API_KEY))
use_native_tools = kwargs.get("use_native_tools", False)
max_tokens = kwargs.get("max_tokens", -1)
@@ -42,7 +42,7 @@ class gtUIHuggingFaceStructureConfig(gtUIBaseConfig):
prompt_model = kwargs.get("prompt_model", None)
temperature = kwargs.get("temperature", 0.7)
stream = kwargs.get("stream", False)
max_attempts = kwargs.get("max_attempts_on_fail", 10)
max_attempts = kwargs.get("max_attempts_on_fail", 2)
use_native_tools = kwargs.get("use_native_tools", False)
api_token = self.getenv(kwargs.get("api_token_env_var", DEFAULT_API_KEY))
configs = {}
@@ -56,7 +56,7 @@ class gtUILMStudioStructureConfig(gtUIBaseConfig):
base_url = kwargs.get("base_url", lmstudio_base_url)
params["base_url"] = f"{base_url}:{port}/v1"
params["temperature"] = kwargs.get("temperature", 0.7)
params["max_attempts"] = kwargs.get("max_attempts_on_fail", 10)
params["max_attempts"] = kwargs.get("max_attempts_on_fail", 2)
params["stream"] = kwargs.get("stream", False)
params["seed"] = kwargs.get("seed", 12341)
params["use_native_tools"] = kwargs.get("use_native_tools", False)
@@ -42,7 +42,7 @@ class gtUIOllamaStructureConfig(gtUIBaseConfig):
params["host"] = f"{base_url}:{port}"
params["stream"] = kwargs.get("stream", False)
params["use_native_tools"] = kwargs.get("use_native_tools", False)
params["max_attempts"] = kwargs.get("max_attempts_on_fail", 10)
params["max_attempts"] = kwargs.get("max_attempts_on_fail", 2)
max_tokens = kwargs.get("max_tokens", -1)
if max_tokens > 0:
params["max_tokens"] = max_tokens
@@ -46,7 +46,7 @@ class gtUIOpenAiCompatibleConfig(gtUIBaseConfig):
params["model"] = kwargs.get("prompt_model", None)
params["base_url"] = kwargs.get("base_url", None)
params["max_attempts"] = kwargs.get("max_attempts_on_fail", 10)
params["max_attempts"] = kwargs.get("max_attempts_on_fail", 2)
params["stream"] = kwargs.get("stream", False)
params["use_native_tools"] = kwargs.get("use_native_tools", False)
api_key_env_var = kwargs.get("api_key_env_var", DEFAULT_API_KEY)
@@ -48,7 +48,7 @@ class gtUIOpenAiStructureConfig(gtUIBaseConfig):
params["model"] = kwargs.get("prompt_model", default_prompt_model)
params["temperature"] = kwargs.get("temperature", 0.7)
params["seed"] = kwargs.get("seed", 12341)
params["max_attempts"] = kwargs.get("max_attempts_on_fail", 10)
params["max_attempts"] = kwargs.get("max_attempts_on_fail", 2)
params["api_key"] = self.getenv(kwargs.get("api_key_env_var", DEFAULT_API_KEY))
params["use_native_tools"] = kwargs.get("use_native_tools", False)
max_tokens = kwargs.get("max_tokens", -1)
+1 -1
View File
@@ -47,7 +47,7 @@ class gtUIBaseConfig:
"seed": ("INT", {"default": 10342349342}),
"max_attempts_on_fail": (
"INT",
{"default": 10, "min": 1, "max": 100},
{"default": 2, "min": 1, "max": 100},
),
# "stream": ([True, False], {"default": False}),
"env": ("ENV", {"default": None}),
+1 -1
View File
@@ -16,7 +16,7 @@ class gtUIBasePromptDriver(gtUIBaseDriver):
"max_attempts_on_fail": (
"INT",
{
"default": 10,
"default": 2,
"min": 1,
"max": 100,
"tooltip": "Maximum attempts on failure",