Chrisyichuan commited on
Commit
622995a
·
1 Parent(s): f22dc3b

add default

Browse files
Files changed (2) hide show
  1. app.py +3 -3
  2. config.py +4 -2
app.py CHANGED
@@ -8,7 +8,7 @@ from pathlib import Path
8
 
9
  from geo_bot import GeoBot, AGENT_PROMPT_TEMPLATE
10
  from benchmark import MapGuesserBenchmark
11
- from config import MODELS_CONFIG, get_data_paths, SUCCESS_THRESHOLD_KM
12
  from langchain_openai import ChatOpenAI
13
  from langchain_anthropic import ChatAnthropic
14
  from langchain_google_genai import ChatGoogleGenerativeAI
@@ -65,13 +65,13 @@ with st.sidebar:
65
  default_dataset = available_datasets[0] if available_datasets else "default"
66
 
67
  dataset_choice = st.selectbox("Dataset", available_datasets, index=0)
68
- model_choice = st.selectbox("Model", list(MODELS_CONFIG.keys()))
69
  steps_per_sample = st.slider("Max Steps", 3, 20, 10)
70
  temperature = st.slider(
71
  "Temperature",
72
  0.0,
73
  2.0,
74
- 0.0,
75
  0.1,
76
  help="Controls randomness in AI responses. 0.0 = deterministic, higher = more creative",
77
  )
 
8
 
9
  from geo_bot import GeoBot, AGENT_PROMPT_TEMPLATE
10
  from benchmark import MapGuesserBenchmark
11
+ from config import MODELS_CONFIG, get_data_paths, SUCCESS_THRESHOLD_KM, DEFAULT_MODEL, DEFAULT_TEMPERATURE
12
  from langchain_openai import ChatOpenAI
13
  from langchain_anthropic import ChatAnthropic
14
  from langchain_google_genai import ChatGoogleGenerativeAI
 
65
  default_dataset = available_datasets[0] if available_datasets else "default"
66
 
67
  dataset_choice = st.selectbox("Dataset", available_datasets, index=0)
68
+ model_choice = st.selectbox("Model", list(MODELS_CONFIG.keys()), index=list(MODELS_CONFIG.keys()).index(DEFAULT_MODEL))
69
  steps_per_sample = st.slider("Max Steps", 3, 20, 10)
70
  temperature = st.slider(
71
  "Temperature",
72
  0.0,
73
  2.0,
74
+ DEFAULT_TEMPERATURE,
75
  0.1,
76
  help="Controls randomness in AI responses. 0.0 = deterministic, higher = more creative",
77
  )
config.py CHANGED
@@ -26,6 +26,10 @@ BENCHMARK_CONFIG = {
26
  # MapCrunch options
27
  MAPCRUNCH_OPTIONS = {}
28
 
 
 
 
 
29
  # Model configurations
30
  MODELS_CONFIG = {
31
  "gpt-4o": {
@@ -70,7 +74,6 @@ MODELS_CONFIG = {
70
  },
71
  }
72
 
73
-
74
  # Data paths - now supports named datasets
75
  def get_data_paths(dataset_name: str = "default"):
76
  """Get data paths for a specific dataset"""
@@ -80,6 +83,5 @@ def get_data_paths(dataset_name: str = "default"):
80
  "results": f"results/{dataset_name}/",
81
  }
82
 
83
-
84
  # Backward compatibility - default paths
85
  DATA_PATHS = get_data_paths("default")
 
26
  # MapCrunch options
27
  MAPCRUNCH_OPTIONS = {}
28
 
29
+ # Default settings
30
+ DEFAULT_MODEL = "gemini-2.5-pro"
31
+ DEFAULT_TEMPERATURE = 1.0
32
+
33
  # Model configurations
34
  MODELS_CONFIG = {
35
  "gpt-4o": {
 
74
  },
75
  }
76
 
 
77
  # Data paths - now supports named datasets
78
  def get_data_paths(dataset_name: str = "default"):
79
  """Get data paths for a specific dataset"""
 
83
  "results": f"results/{dataset_name}/",
84
  }
85
 
 
86
  # Backward compatibility - default paths
87
  DATA_PATHS = get_data_paths("default")