Hemang Thakur commited on
Commit
e67814c
·
1 Parent(s): d8d270b

rebuild chat google gemini model

Browse files
Files changed (1) hide show
  1. src/utils/api_key_manager.py +3 -0
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: