Add title to the base url field (#79)

airtai · Aug 26, 2024 · 2f2a85e · 2f2a85e
1 parent 24932e4
commit 2f2a85e
Show file tree

Hide file tree

Showing 8 changed files with 14 additions and 14 deletions.
diff --git a/fastagency/studio/models/llms/anthropic.py b/fastagency/studio/models/llms/anthropic.py
@@ -69,9 +69,9 @@ class Anthropic(Model):
         ),
     ]
 
-    base_url: Annotated[URL, Field(description="The base URL of the Anthropic API")] = (
-        URL(url="https://api.anthropic.com/v1")
-    )
+    base_url: Annotated[
+        URL, Field(title="Base URL", description="The base URL of the Anthropic API")
+    ] = URL(url="https://api.anthropic.com/v1")
 
     api_type: Annotated[
         Literal["anthropic"],

diff --git a/fastagency/studio/models/llms/azure.py b/fastagency/studio/models/llms/azure.py
@@ -67,7 +67,7 @@ class AzureOAI(Model):
     ]
 
     base_url: Annotated[
-        URL, Field(description="The base URL of the Azure OpenAI API")
+        URL, Field(title="Base URL", description="The base URL of the Azure OpenAI API")
     ] = UrlModel(url="https://{your-resource-name}.openai.azure.com").url
 
     api_type: Annotated[

diff --git a/fastagency/studio/models/llms/openai.py b/fastagency/studio/models/llms/openai.py
@@ -82,9 +82,9 @@ class OpenAI(Model):
         ),
     ]
 
-    base_url: Annotated[URL, Field(description="The base URL of the OpenAI API")] = URL(
-        url="https://api.openai.com/v1"
-    )
+    base_url: Annotated[
+        URL, Field(title="Base URL", description="The base URL of the OpenAI API")
+    ] = URL(url="https://api.openai.com/v1")
 
     api_type: Annotated[
         Literal["openai"],

diff --git a/fastagency/studio/models/llms/together.py b/fastagency/studio/models/llms/together.py
@@ -150,9 +150,9 @@ class TogetherAI(Model):
         Field(title="API Key", description="The API Key from Together.ai"),
     ]
 
-    base_url: Annotated[URL, Field(description="The base URL of the OpenAI API")] = URL(
-        url="https://api.together.xyz/v1"
-    )
+    base_url: Annotated[
+        URL, Field(title="Base URL", description="The base URL of the OpenAI API")
+    ] = URL(url="https://api.together.xyz/v1")
 
     api_type: Annotated[
         Literal["togetherai"],

diff --git a/tests/studio/models/llms/test_anthropic.py b/tests/studio/models/llms/test_anthropic.py
@@ -113,7 +113,7 @@ def test_anthropic_model_schema(self) -> None:
                     "format": "uri",
                     "maxLength": 2083,
                     "minLength": 1,
-                    "title": "Base Url",
+                    "title": "Base URL",
                     "type": "string",
                 },
                 "api_type": {

diff --git a/tests/studio/models/llms/test_azure.py b/tests/studio/models/llms/test_azure.py
@@ -138,7 +138,7 @@ def test_azure_model_schema(self) -> None:
                     "format": "uri",
                     "maxLength": 2083,
                     "minLength": 1,
-                    "title": "Base Url",
+                    "title": "Base URL",
                     "type": "string",
                 },
                 "api_type": {

diff --git a/tests/studio/models/llms/test_openai.py b/tests/studio/models/llms/test_openai.py
@@ -152,7 +152,7 @@ def test_openai_schema(self) -> None:
                     "format": "uri",
                     "maxLength": 2083,
                     "minLength": 1,
-                    "title": "Base Url",
+                    "title": "Base URL",
                     "type": "string",
                 },
                 "api_type": {

diff --git a/tests/studio/models/llms/test_together.py b/tests/studio/models/llms/test_together.py
@@ -143,7 +143,7 @@ def test_togetherai_schema(self) -> None:
                     "format": "uri",
                     "maxLength": 2083,
                     "minLength": 1,
-                    "title": "Base Url",
+                    "title": "Base URL",
                     "type": "string",
                 },
                 "api_type": {