Merge pull request #16 from technologiestiftung/staging

feat: more details for model endpoint
technologiestiftung · Aug 27, 2024 · 4bfb897 · 4bfb897
2 parents 66fc344 + 3ce3a25
commit 4bfb897
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/src/controllers/model-controller.ts b/src/controllers/model-controller.ts
@@ -11,20 +11,31 @@ export const getModels = (req: Request, res: Response<ModelResponse>) => {
         provider: ModelProvider.OpenAI,
         isGdprCompliant: false,
         contextSize: 128000,
+        isOpenSource: false,
+        serverLocation: "USA",
+        description: "Aktuelles Modell von OpenAI, gehostet von OpenAI.",
       },
       {
         identifier: "azure-gpt-4o-mini",
         baseModelName: "gpt-4o-mini",
         provider: ModelProvider.Azure,
         isGdprCompliant: true,
         contextSize: 128000,
+        isOpenSource: false,
+        serverLocation: "Schweden",
+        description:
+          "Aktuelles Modell von OpenAI, datenschutzkonform gehostet von Microsoft Azure.",
       },
       {
         identifier: "citylab-macstudio-llama-3.1",
         baseModelName: "llama3.1",
         provider: ModelProvider.Ollama,
         isGdprCompliant: true,
         contextSize: 128000,
+        isOpenSource: true,
+        serverLocation: "Berlin",
+        description:
+          "Open Source - Modell von Meta, datenschutzkonform gehostet im CityLAB Berlin.",
       },
     ],
     parameters: LLM_PARAMETERS,

diff --git a/src/types/model-types.ts b/src/types/model-types.ts
@@ -13,6 +13,9 @@ export interface Model {
   provider: ModelProvider;
   isGdprCompliant: boolean;
   contextSize: number;
+  isOpenSource: boolean;
+  serverLocation: string;
+  description: string;
 }
 
 export interface ModelResponse {