Apply suggestions from code review

Co-authored-by: Lucie Milan <[email protected]>
Kong · Dec 18, 2024 · cc08269 · cc08269
1 parent 08f478b
commit cc08269
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 12 deletions.
diff --git a/app/_includes/md/plugins-hub/ai-custom-model-advanced.md b/app/_includes/md/plugins-hub/ai-custom-model-advanced.md
@@ -15,9 +15,9 @@ config:
     auth:
       header_name: "Authorization"
       header_value: "Bearer <openai_key>"
-  - model:
+    model:
       name: custom_model_name
-      provider: openai|azure|anthropic|cohere|mistral|llama2|gemini|bedrock
+      provider: openai|azure|anthropic|cohere|mistral|llama2|gemini|bedrock|huggingface
       options:
         upstream_url: http://localhost:8000/vi/chat/completions
 targets:

diff --git a/app/_includes/md/plugins-hub/ai-custom-model.md b/app/_includes/md/plugins-hub/ai-custom-model.md
@@ -10,16 +10,15 @@ You can configure the AI Proxy Plugin using a custom model of your choice by set
 plugin: kong-inc/ai-proxy-advanced
 name: ai-advanced
 config:
-  targets:
-  - route_type: "llm/v1/chat"
-    auth:
-      header_name: "Authorization"
-      header_value: "Bearer <openai_key>"
-  - model:
-      name: custom_model_name
-      provider: openai|azure|anthropic|cohere|mistral|llama2|gemini|bedrock
-      options:
-        upstream_url: http://localhost:8000/vi/chat/completions
+  route_type: "llm/v1/chat"
+  auth:
+    header_name: "Authorization"
+    header_value: "Bearer <openai_key>"
+  model:
+    name: custom_model_name
+    provider: openai|azure|anthropic|cohere|mistral|llama2|gemini|bedrock|huggingface
+    options:
+      upstream_url: http://localhost:8000/vi/chat/completions
 targets:
   - route
 formats: