harvesthealth commited on
Commit
2b64364
·
verified ·
1 Parent(s): fc449b3

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. app.py +10 -5
app.py CHANGED
@@ -105,34 +105,39 @@ os.makedirs(LOG_DIR, exist_ok=True)
105
  HELMHOLTZ_BASE_URL = "https://api.helmholtz-blablador.fz-juelich.de/v1"
106
 
107
  # Initialize LLMs (Helmholtz Blablador)
108
- api_key = os.environ.get("BLABLADOR_API_KEY")
 
109
 
110
  chat_llm = ChatOpenAI(
111
  model="alias-large",
112
  base_url=HELMHOLTZ_BASE_URL,
113
  api_key=api_key,
114
- max_tokens=2048
 
115
  )
116
 
117
  code_llm = ChatOpenAI(
118
  model="alias-code",
119
  base_url=HELMHOLTZ_BASE_URL,
120
  api_key=api_key,
121
- max_tokens=1024
 
122
  )
123
 
124
  fast_llm = ChatOpenAI(
125
  model="alias-fast",
126
  base_url=HELMHOLTZ_BASE_URL,
127
  api_key=api_key,
128
- max_tokens=512
 
129
  )
130
 
131
  huge_llm = ChatOpenAI(
132
  model="alias-huge",
133
  base_url=HELMHOLTZ_BASE_URL,
134
  api_key=api_key,
135
- max_tokens=1024
 
136
  )
137
 
138
  # Memory Stores
 
105
  HELMHOLTZ_BASE_URL = "https://api.helmholtz-blablador.fz-juelich.de/v1"
106
 
107
  # Initialize LLMs (Helmholtz Blablador)
108
+ # We use a placeholder if the API key is missing to allow the app to start and show status.
109
+ api_key = os.environ.get("BLABLADOR_API_KEY") or "EMPTY_KEY"
110
 
111
  chat_llm = ChatOpenAI(
112
  model="alias-large",
113
  base_url=HELMHOLTZ_BASE_URL,
114
  api_key=api_key,
115
+ max_tokens=2048,
116
+ disable_streaming=True
117
  )
118
 
119
  code_llm = ChatOpenAI(
120
  model="alias-code",
121
  base_url=HELMHOLTZ_BASE_URL,
122
  api_key=api_key,
123
+ max_tokens=1024,
124
+ disable_streaming=True
125
  )
126
 
127
  fast_llm = ChatOpenAI(
128
  model="alias-fast",
129
  base_url=HELMHOLTZ_BASE_URL,
130
  api_key=api_key,
131
+ max_tokens=512,
132
+ disable_streaming=True
133
  )
134
 
135
  huge_llm = ChatOpenAI(
136
  model="alias-huge",
137
  base_url=HELMHOLTZ_BASE_URL,
138
  api_key=api_key,
139
+ max_tokens=1024,
140
+ disable_streaming=True
141
  )
142
 
143
  # Memory Stores