charieswei
/
TradingAgents
mirror of https://github.com/TauricResearch/TradingAgents.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

update select_language()

This commit is contained in:
Jeffrey Chu 2025-10-23 18:06:57 +08:00
parent ea944f860c
commit 349ac8f2fe
1 changed files with 41 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

62
cli/utils.py
View File

@ -1,5 +1,6 @@
from typing import List
import questionary import questionary
from typing import List, Optional, Tuple, Dict
from cli.models import AnalystType from cli.models import AnalystType
@ -48,7 +49,7 @@ def get_analysis_date() -> str:
date = questionary.text( date = questionary.text(
"Enter the analysis date (YYYY-MM-DD):", "Enter the analysis date (YYYY-MM-DD):",
validate=lambda x: validate_date(x.strip()) validate=lambda x: validate_date(x.strip())
or "Please enter a valid date in YYYY-MM-DD format.", or "Please enter a valid date in YYYY-MM-DD format.",
style=questionary.Style( style=questionary.Style(
[ [
("text", "fg:green"), ("text", "fg:green"),
@ -146,8 +147,10 @@ def select_shallow_thinking_agent(provider) -> str:
], ],
"openrouter": [ "openrouter": [
("Meta: Llama 4 Scout", "meta-llama/llama-4-scout:free"), ("Meta: Llama 4 Scout", "meta-llama/llama-4-scout:free"),
("Meta: Llama 3.3 8B Instruct - A lightweight and ultra-fast variant of Llama 3.3 70B", "meta-llama/llama-3.3-8b-instruct:free"), ("Meta: Llama 3.3 8B Instruct - A lightweight and ultra-fast variant of Llama 3.3 70B",
("google/gemini-2.0-flash-exp:free - Gemini Flash 2.0 offers a significantly faster time to first token", "google/gemini-2.0-flash-exp:free"), "meta-llama/llama-3.3-8b-instruct:free"),
("google/gemini-2.0-flash-exp:free - Gemini Flash 2.0 offers a significantly faster time to first token",
"google/gemini-2.0-flash-exp:free"),
], ],
"ollama": [ "ollama": [
("llama3.1 local", "llama3.1"), ("llama3.1 local", "llama3.1"),
@ -212,7 +215,8 @@ def select_deep_thinking_agent(provider) -> str:
], ],
"openrouter": [ "openrouter": [
("DeepSeek V3 - a 685B-parameter, mixture-of-experts model", "deepseek/deepseek-chat-v3-0324:free"), ("DeepSeek V3 - a 685B-parameter, mixture-of-experts model", "deepseek/deepseek-chat-v3-0324:free"),
("Deepseek - latest iteration of the flagship chat model family from the DeepSeek team.", "deepseek/deepseek-chat-v3-0324:free"), ("Deepseek - latest iteration of the flagship chat model family from the DeepSeek team.",
"deepseek/deepseek-chat-v3-0324:free"),
], ],
"ollama": [ "ollama": [
("llama3.1 local", "llama3.1"), ("llama3.1 local", "llama3.1"),
@ -222,7 +226,7 @@ def select_deep_thinking_agent(provider) -> str:
("grok-4-fast-reasoning", "grok-4-fast-reasoning"), ("grok-4-fast-reasoning", "grok-4-fast-reasoning"),
] ]
} }
choice = questionary.select( choice = questionary.select(
"Select Your [Deep-Thinking LLM Engine]:", "Select Your [Deep-Thinking LLM Engine]:",
choices=[ choices=[
@ -245,18 +249,19 @@ def select_deep_thinking_agent(provider) -> str:
return choice return choice
def select_llm_provider() -> tuple[str, str]: def select_llm_provider() -> tuple[str, str]:
"""Select the OpenAI api url using interactive selection.""" """Select the OpenAI api url using interactive selection."""
# Define OpenAI api options with their corresponding endpoints # Define OpenAI api options with their corresponding endpoints
BASE_URLS = [ BASE_URLS = [
("XAI", "https://api.x.ai/v1"), ("XAI", "https://api.x.ai/v1"),
("OpenAI", "https://api.openai.com/v1"), ("OpenAI", "https://api.openai.com/v1"),
("Anthropic", "https://api.anthropic.com/"), ("Anthropic", "https://api.anthropic.com/"),
("Google", "https://generativelanguage.googleapis.com/v1"), ("Google", "https://generativelanguage.googleapis.com/v1"),
("Openrouter", "https://openrouter.ai/api/v1"), ("Openrouter", "https://openrouter.ai/api/v1"),
("Ollama", "http://localhost:11434/v1"), ("Ollama", "http://localhost:11434/v1"),
] ]
choice = questionary.select( choice = questionary.select(
"Select your LLM Provider:", "Select your LLM Provider:",
choices=[ choices=[
@ -272,27 +277,42 @@ def select_llm_provider() -> tuple[str, str]:
] ]
), ),
).ask() ).ask()
if choice is None: if choice is None:
console.print("\n[red]no OpenAI backend selected. Exiting...[/red]") console.print("\n[red]no OpenAI backend selected. Exiting...[/red]")
exit(1) exit(1)
display_name, url = choice display_name, url = choice
print(f"You selected: {display_name}\tURL: {url}") print(f"You selected: {display_name}\tURL: {url}")
return display_name, url return display_name, url
def select_language() -> str: def select_language() -> int:
"""Select output language for agent responses.""" """Select output language for agent responses."""
choices = [ LANGUAGE_OPTIONS = [
questionary.Choice("English (default)", "en"), ("English", "en"),
questionary.Choice("Traditional Chinese", "zh-tw"), ("Traditional Chinese", "zh-tw"),
questionary.Choice("Simplified Chinese", "zh-cn"), ("Simplified Chinese", "zh-cn"),
] ]
return questionary.select(
choice = questionary.select(
"Select Output Language for Agents:", "Select Output Language for Agents:",
choices=choices, choices=[
default="en", questionary.Choice(display, value=value) for display, value in LANGUAGE_OPTIONS
style=questionary.Style([("selected", "fg:cyan noinherit")]) ],
).ask() or "en" # Default to 'en' if None instruction="\n- Use arrow keys to navigate\n- Press Enter to select",
style=questionary.Style(
[
("selected", "fg:yellow noinherit"),
("highlighted", "fg:yellow noinherit"),
("pointer", "fg:yellow noinherit"),
]
),
).ask()
if choice is None:
console.print("\n[red]No langauge selected. Exiting...[/red]")
exit(1)
return choice