2
0
visuddhinanda 4 сар өмнө
parent
commit
4cedd1f4f9

+ 10 - 2
api-v8/app/Services/AIModelService.php

@@ -16,13 +16,21 @@ class AIModelService
         return AiModelResource::collection(resource: $result);
     }
 
-    public function getSysModels()
+    public function getSysModels($type = null)
     {
-        $types = ['wbw', 'chat'];
+        if (empty($type)) {
+            $types = ['wbw', 'chat', 'summarize'];
+        } else {
+            $types = [$type];
+        }
+
         $sysModels = [];
         foreach ($types as $key => $type) {
             $sysModels[$type] =  $this->getModelsById(RedisClusters::get('/ai/model/system/' . $type) ?? []);
         }
+        if (!empty($type)) {
+            return $sysModels[$type];
+        }
         return $sysModels;
     }
 }