Merge pull request #3 from BaizeAI/fix/org-name

fix: org name

Author: kebe7jun

hack/import_db.py

@@ -56,15 +56,52 @@ def get_source(model_path):
         }
     }
 
+class Provider:
+    id = ""
+    _id = ""
+    name = ""
+    zhCN_name = ""
+
+    def __init__(self, provider_str):
+        provider = json.loads(provider_str)
+        name = provider['enUS']
+        provider_id = str.lower(name)
+        self.id = provider_id
+        self.name = name
+        self.zhCN_name = provider['zhCN']
+    
+    def to_map(self) -> dict:
+        return {
+            "id": self.id,
+            "name": {
+                "enUS": self.name,
+                "zhCN": self.zhCN_name
+            }
+        }
+    
+    def with_suggest(self, sources, orgs):
+        org = get_org_from_source(sources)
+        suggest = orgs.get(org)
+        if not suggest:
+            return self
+        
+        if "enUS" in suggest:
+            self.name = suggest["enUS"]
+            self.id = str.lower(self.name)
+        
+        if "zhCN" in suggest:
+            self.zhCN_name = suggest["zhCN"]
+        
+        return self
 
 def create_yaml(model_data):
     model_id = model_data['model_id']
     model_name = model_data['model_name']
     model_description = yaml.safe_load(model_data['model_description'])
     resource_requirements = parse_resource_requirements(model_data['resources_requirements'], model_data['model_image'], model_data['custom_deploy_args'])
-    provider_name = json.loads(model_data['provider'])
-    provider_id = str.lower(provider_name['enUS'])
-    model_path = f'{provider_id}/{model_id}'
+
+    provider = Provider(model_data['provider'])
+    model_path = f'{provider.id}/{model_id}'
     source = get_source(model_path)
 
     yaml_content = {
@@ -84,19 +121,14 @@ def create_yaml(model_data):
                     'src': model_data['model_avatar'],
                     'type': 'image/svg'
                 },
-                'provider': {
-                    'id': provider_id,
-                    'name': provider_name,
-                },
+                'provider': provider.with_suggest(source, orgs).to_map(),
                 'tags': json.loads(model_data['model_support_feature'])
             }
         }
     }
 
     return yaml_content
 
-
-
 def use_suggests(platform, model_path):
     _, model_name = get_org_and_model_id(model_path)
     suggest = suggests.get(model_name)
@@ -143,6 +175,12 @@ def links(source):
     return []
 
 
+def get_org_from_source(source):
+    org, _ = get_org_and_model_id(source['modelscope']['name'])
+    if org == "LLM-Research":   # LLM-Research is a common org name
+        org, _ = get_org_and_model_id(source['huggingface']['name'])
+    return org
+
 # {
 #     "Code": 200,
 #     "Data": {
@@ -164,9 +202,7 @@ def links(source):
 #     "Success": true
 # }
 def add_modelscope_about(source):
-    org, _ = get_org_and_model_id(source['modelscope']['name'])
-    if org == "LLM-Research":   # LLM-Research is a common org name
-        org, _ = get_org_and_model_id(source['huggingface']['name'])
+    org = get_org_from_source(source)
 
     if org in orgs:
         return {

hack/orgs.json

@@ -3,16 +3,20 @@
         "link": "https://github.com/baichuan-inc"
     },
     "deepseek-ai": {
-        "link": "https://www.deepseek.com/"
+        "link": "https://www.deepseek.com/",
+        "zhCN": "\u6df1\u5ea6\u6c42\u7d22"
     },
     "ZhipuAI": {
-        "link": "https://www.zhipu.ai"
+        "link": "https://www.zhipu.ai",
+        "zhCN": "\u667a\u8c31AI",
+        "enUS": "ZhipuAI"
     },
     "meta-llama": {
         "link": "https://ai.meta.com/llama/"
     },
     "MiniMaxAI": {
-        "link": "https://www.minimaxi.com/"
+        "link": "https://www.minimaxi.com/",
+        "zhCN": "\u7a00\u5b87\u79d1\u6280"
     },
     "microsoft": {
         "link": "https://www.microsoft.com/en-us/research/"

models/deepseek/deepseek-r1-distill-qwen-1.5b/metadata.yaml

@@ -29,7 +29,7 @@ spec:
       id: deepseek
       name:
         enUS: DeepSeek
-        zhCN: DeepSeek
+        zhCN: 深度求索
     tags:
     - TEXT_GENERATION
   source:

models/deepseek/deepseek-r1-distill-qwen-14b/metadata.yaml

@@ -33,7 +33,7 @@ spec:
       id: deepseek
       name:
         enUS: DeepSeek
-        zhCN: DeepSeek
+        zhCN: 深度求索
     tags:
     - TEXT_GENERATION
   source:

models/deepseek/deepseek-r1-distill-qwen-32b/metadata.yaml

@@ -35,7 +35,7 @@ spec:
       id: deepseek
       name:
         enUS: DeepSeek
-        zhCN: DeepSeek
+        zhCN: 深度求索
     tags:
     - TEXT_GENERATION
   source:

models/deepseek/deepseek-r1/metadata.yaml

@@ -31,7 +31,7 @@ spec:
       id: deepseek
       name:
         enUS: DeepSeek
-        zhCN: DeepSeek
+        zhCN: 深度求索
     tags:
     - TEXT_GENERATION
   source:

models/deepseek/deepseek-v3/metadata.yaml

@@ -31,7 +31,7 @@ spec:
       id: deepseek
       name:
         enUS: DeepSeek
-        zhCN: DeepSeek
+        zhCN: 深度求索
     tags:
     - TEXT_GENERATION
   source:

models/minimax/minimax-text-01/metadata.yaml

@@ -29,7 +29,7 @@ spec:
       id: minimax
       name:
         enUS: MiniMax
-        zhCN: MiniMax
+        zhCN: 稀宇科技
     tags:
     - TEXT_GENERATION
   source:

models/glm/glm-4-9b-chat/metadata.yaml …dels/zhipuai/glm-4-9b-chat/metadata.yamlmodels/glm/glm-4-9b-chat/metadata.yaml renamed to models/zhipuai/glm-4-9b-chat/metadata.yaml models/glm/glm-4-9b-chat/metadata.yaml renamed to models/zhipuai/glm-4-9b-chat/metadata.yaml

@@ -26,10 +26,10 @@ spec:
     - description: About
       url: https://www.zhipu.ai
     provider:
-      id: glm
+      id: zhipuai
       name:
-        enUS: GLM
-        zhCN: GLM
+        enUS: ZhipuAI
+        zhCN: 智谱AI
     tags:
     - TEXT_GENERATION
   source: