Spaces:
Running
Running
Hemang Thakur
commited on
Commit
·
e67814c
1
Parent(s):
d8d270b
rebuild chat google gemini model
Browse files
src/utils/api_key_manager.py
CHANGED
@@ -15,6 +15,9 @@ from google.api_core.exceptions import ResourceExhausted, BadRequest, InvalidArg
|
|
15 |
from tenacity import retry, wait_random_exponential, stop_after_attempt, retry_if_exception_type
|
16 |
import asyncio
|
17 |
|
|
|
|
|
|
|
18 |
ModelProvider = Literal["openai", "anthropic", "google", "xai"]
|
19 |
|
20 |
class APIKeyManager:
|
|
|
15 |
from tenacity import retry, wait_random_exponential, stop_after_attempt, retry_if_exception_type
|
16 |
import asyncio
|
17 |
|
18 |
+
# Tell Pydantic to finish wiring up this dynamic model
|
19 |
+
ChatGoogleGenerativeAI.model_rebuild()
|
20 |
+
|
21 |
ModelProvider = Literal["openai", "anthropic", "google", "xai"]
|
22 |
|
23 |
class APIKeyManager:
|