add Moonshot, debug my_llm (#126)

2024-03-15 18:59:00 +08:00
parent de09b0e1a4
commit 2447f95629
6 changed files with 76 additions and 48 deletions
--- a/api/db/db_models.py
+++ b/api/db/db_models.py
@@ -429,7 +429,7 @@ class LLMFactories(DataBaseModel):

 class LLM(DataBaseModel):
    # LLMs dictionary
-    llm_name = CharField(max_length=128, null=False, help_text="LLM name", index=True)
+    llm_name = CharField(max_length=128, null=False, help_text="LLM name", index=True, primary_key=True)
    model_type = CharField(max_length=128, null=False, help_text="LLM, Text Embedding, Image2Text, ASR")
    fid = CharField(max_length=128, null=False, help_text="LLM factory id")
    max_tokens = IntegerField(default=0)