Fix int value (#256)

aigc-apps · Oct 25, 2024 · 90ec8eb · 90ec8eb
1 parent 12a2862
commit 90ec8eb
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/pai_rag/integrations/llms/pai/llm_config.py b/src/pai_rag/integrations/llms/pai/llm_config.py
@@ -1,5 +1,5 @@
 from typing import Literal
-from pydantic import BaseModel
+from pydantic import BaseModel, ConfigDict
 from enum import Enum
 from llama_index.core.constants import DEFAULT_TEMPERATURE
 
@@ -159,6 +159,8 @@ class PaiEasLlmConfig(PaiBaseLlmConfig):
     token: str
     model: str = "default"
 
+    model_config = ConfigDict(coerce_numbers_to_str=True)
+
 
 class DashScopeMultiModalLlmConfig(DashScopeLlmConfig):
     model: str = "qwen-vl-max"