refactor: clean up imports
This commit is contained in:
parent
87e967a4a2
commit
f8ff71f4c7
24
cli/utils.py
24
cli/utils.py
|
|
@ -1,8 +1,14 @@
|
||||||
import questionary
|
import questionary
|
||||||
from typing import List, Optional, Tuple, Dict
|
import re
|
||||||
|
from datetime import datetime
|
||||||
|
from rich.console import Console
|
||||||
|
from typing import List
|
||||||
|
from urllib.parse import urlparse
|
||||||
|
|
||||||
from cli.models import AnalystType
|
from cli.models import AnalystType
|
||||||
|
|
||||||
|
console = Console()
|
||||||
|
|
||||||
ANALYST_ORDER = [
|
ANALYST_ORDER = [
|
||||||
("Market Analyst", AnalystType.MARKET),
|
("Market Analyst", AnalystType.MARKET),
|
||||||
("Social Media Analyst", AnalystType.SOCIAL),
|
("Social Media Analyst", AnalystType.SOCIAL),
|
||||||
|
|
@ -33,8 +39,6 @@ def get_ticker() -> str:
|
||||||
|
|
||||||
def get_analysis_date() -> str:
|
def get_analysis_date() -> str:
|
||||||
"""Prompt the user to enter a date in YYYY-MM-DD format."""
|
"""Prompt the user to enter a date in YYYY-MM-DD format."""
|
||||||
import re
|
|
||||||
from datetime import datetime
|
|
||||||
|
|
||||||
def validate_date(date_str: str) -> bool:
|
def validate_date(date_str: str) -> bool:
|
||||||
if not re.match(r"^\d{4}-\d{2}-\d{2}$", date_str):
|
if not re.match(r"^\d{4}-\d{2}-\d{2}$", date_str):
|
||||||
|
|
@ -275,8 +279,6 @@ def select_shallow_thinking_agent(provider) -> str:
|
||||||
default_model="gpt-4o-mini"
|
default_model="gpt-4o-mini"
|
||||||
)
|
)
|
||||||
except ValueError as e:
|
except ValueError as e:
|
||||||
from rich.console import Console
|
|
||||||
console = Console()
|
|
||||||
console.print(f"\n[red]Error: {e}[/red]")
|
console.print(f"\n[red]Error: {e}[/red]")
|
||||||
exit(1)
|
exit(1)
|
||||||
|
|
||||||
|
|
@ -299,8 +301,6 @@ def select_shallow_thinking_agent(provider) -> str:
|
||||||
).ask()
|
).ask()
|
||||||
|
|
||||||
if choice is None:
|
if choice is None:
|
||||||
from rich.console import Console
|
|
||||||
console = Console()
|
|
||||||
console.print(
|
console.print(
|
||||||
"\n[red]No shallow thinking llm engine selected. Exiting...[/red]"
|
"\n[red]No shallow thinking llm engine selected. Exiting...[/red]"
|
||||||
)
|
)
|
||||||
|
|
@ -321,8 +321,6 @@ def select_deep_thinking_agent(provider) -> str:
|
||||||
default_model="o4-mini"
|
default_model="o4-mini"
|
||||||
)
|
)
|
||||||
except ValueError as e:
|
except ValueError as e:
|
||||||
from rich.console import Console
|
|
||||||
console = Console()
|
|
||||||
console.print(f"\n[red]Error: {e}[/red]")
|
console.print(f"\n[red]Error: {e}[/red]")
|
||||||
exit(1)
|
exit(1)
|
||||||
|
|
||||||
|
|
@ -345,8 +343,6 @@ def select_deep_thinking_agent(provider) -> str:
|
||||||
).ask()
|
).ask()
|
||||||
|
|
||||||
if choice is None:
|
if choice is None:
|
||||||
from rich.console import Console
|
|
||||||
console = Console()
|
|
||||||
console.print("\n[red]No deep thinking llm engine selected. Exiting...[/red]")
|
console.print("\n[red]No deep thinking llm engine selected. Exiting...[/red]")
|
||||||
exit(1)
|
exit(1)
|
||||||
|
|
||||||
|
|
@ -364,8 +360,6 @@ def validate_custom_url(url: str) -> str:
|
||||||
Raises:
|
Raises:
|
||||||
ValueError: If the URL is invalid or malformed
|
ValueError: If the URL is invalid or malformed
|
||||||
"""
|
"""
|
||||||
import re
|
|
||||||
from urllib.parse import urlparse
|
|
||||||
|
|
||||||
if not url:
|
if not url:
|
||||||
return ""
|
return ""
|
||||||
|
|
@ -406,7 +400,6 @@ def get_custom_provider_info() -> tuple[str, str] | None:
|
||||||
"""
|
"""
|
||||||
import os
|
import os
|
||||||
from urllib.parse import urlparse
|
from urllib.parse import urlparse
|
||||||
from rich.console import Console
|
|
||||||
|
|
||||||
custom_url = os.getenv("CUSTOM_BASE_URL")
|
custom_url = os.getenv("CUSTOM_BASE_URL")
|
||||||
custom_api_key = os.getenv("CUSTOM_API_KEY")
|
custom_api_key = os.getenv("CUSTOM_API_KEY")
|
||||||
|
|
@ -418,7 +411,6 @@ def get_custom_provider_info() -> tuple[str, str] | None:
|
||||||
hostname = parsed.netloc
|
hostname = parsed.netloc
|
||||||
return f"Custom ({hostname})", validated_url
|
return f"Custom ({hostname})", validated_url
|
||||||
except ValueError as e:
|
except ValueError as e:
|
||||||
console = Console()
|
|
||||||
console.print(f"[red]Error: {e}[/red]")
|
console.print(f"[red]Error: {e}[/red]")
|
||||||
exit(1)
|
exit(1)
|
||||||
|
|
||||||
|
|
@ -459,8 +451,6 @@ def select_llm_provider() -> tuple[str, str]:
|
||||||
).ask()
|
).ask()
|
||||||
|
|
||||||
if choice is None:
|
if choice is None:
|
||||||
from rich.console import Console
|
|
||||||
console = Console()
|
|
||||||
console.print("\n[red]No LLM provider selected. Exiting...[/red]")
|
console.print("\n[red]No LLM provider selected. Exiting...[/red]")
|
||||||
exit(1)
|
exit(1)
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue