Spaces:
Paused
Paused
Upload folder using huggingface_hub
Browse files
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 |
-
|
|
|
|
| 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
|