modify default model value by doc (#334)

dataelement · Feb 7, 2024 · c87527c · c87527c
2 parents 4906e31 + 228f684
commit c87527c
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/src/backend/bisheng/database/models/template.py b/src/backend/bisheng/database/models/template.py
@@ -20,7 +20,7 @@ class TemplateSkillBase(SQLModelSerializable):
                          nullable=False,
                          server_default=text('CURRENT_TIMESTAMP'),
                          onupdate=text('CURRENT_TIMESTAMP')))
-    guide_word: Optional[str] = Field(index=False)
+    guide_word: Optional[str] = Field(index=False, max_length=1000)
 
 
 class Template(TemplateSkillBase, table=True):

diff --git a/src/bisheng-langchain/bisheng_langchain/chat_models/host_llm.py b/src/bisheng-langchain/bisheng_langchain/chat_models/host_llm.py
@@ -540,8 +540,8 @@ class HostYuanChat(BaseHostChatLLM):
     # use custom llm chat api, api should compatiable with openai definition
     model_name: str = Field('Yuan2-2B-Janus-hf', alias='model')
 
-    temperature: float = 0.1
-    top_p: float = 0.1
+    temperature: float = 1
+    top_p: float = 0.9
     max_tokens: int = 4096
     host_base_url: str
 
@@ -554,8 +554,8 @@ class HostYiChat(BaseHostChatLLM):
     # use custom llm chat api, api should compatiable with openai definition
     model_name: str = Field('Yi-34B-Chat', alias='model')
 
-    temperature: float = 0.1
-    top_p: float = 0.1
+    temperature: float = 0.6
+    top_p: float = 0.8
     max_tokens: int = 4096
     host_base_url: str