Added Link attribute in embedder and llm python files

cheshire-cat-ai · Oct 29, 2023 · f74d7a4 · f74d7a4
1 parent 464bbe3
commit f74d7a4
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/core/cat/factory/embedder.py b/core/cat/factory/embedder.py
@@ -34,6 +34,7 @@ class EmbedderFakeConfig(EmbedderSettings):
         json_schema_extra = {
             "humanReadableName": "Default Embedder",
             "description": "Configuration for default embedder. It just outputs random numbers.",
+            "link": ""
         }
     )
 
@@ -46,6 +47,7 @@ class EmbedderDumbConfig(EmbedderSettings):
         json_schema_extra = {
             "humanReadableName": "Dumb Embedder",
             "description": "Configuration for default embedder. It encodes the pairs of characters",
+            "link": ""
         }
     )
 
@@ -58,6 +60,7 @@ class EmbedderLlamaCppConfig(EmbedderSettings):
         json_schema_extra = {
             "humanReadableName": "Self-hosted llama-cpp-python embedder",
             "description": "Self-hosted llama-cpp-python embedder",
+            "link" : ""
         }
     )
 
@@ -71,6 +74,7 @@ class EmbedderOpenAIConfig(EmbedderSettings):
         json_schema_extra = {
             "humanReadableName": "OpenAI Embedder",
             "description": "Configuration for OpenAI embeddings",
+            "link": "https://platform.openai.com/docs/models/overview"
         }
     )
 
@@ -90,6 +94,7 @@ class EmbedderAzureOpenAIConfig(EmbedderSettings):
         json_schema_extra = {
             "humanReadableName": "Azure OpenAI Embedder",
             "description": "Configuration for Azure OpenAI embeddings",
+            "link": "https://azure.microsoft.com/en-us/products/ai-services/openai-service"
         }
     )
 
@@ -103,6 +108,7 @@ class EmbedderCohereConfig(EmbedderSettings):
         json_schema_extra = {
             "humanReadableName": "Cohere Embedder",
             "description": "Configuration for Cohere embeddings",
+            "link": "https://docs.cohere.com/docs/models"
         }
     )
 
@@ -116,6 +122,7 @@ class EmbedderHuggingFaceHubConfig(EmbedderSettings):
         json_schema_extra = {
             "humanReadableName": "HuggingFace Hub Embedder",
             "description": "Configuration for HuggingFace Hub embeddings",
+            "link": "https://huggingface.co/models"
         }
     )
 

diff --git a/core/cat/factory/llm.py b/core/cat/factory/llm.py
@@ -90,6 +90,7 @@ class LLMLlamaCppConfig(LLMSettings):
         json_schema_extra = {
             "humanReadableName": "Self-hosted llama-cpp-python",
             "description": "Self-hosted llama-cpp-python compatible LLM",
+            "link": ""
         }
     )