change max attempts to 2 from 10
This commit is contained in:
@@ -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)
|
||||
|
||||
@@ -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}),
|
||||
|
||||
@@ -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",
|
||||
|
||||
Reference in New Issue
Block a user