Skip to content

Commit d5f09c0

Browse files
authored
Merge pull request #1628 from weaviate/fix/urls-in-generative-config-runtime
Fix accidental adding of `/` to all `AnyHttpUrl` models in grpc gen.conf.
2 parents 026364c + 0bb3c32 commit d5f09c0

2 files changed

Lines changed: 16 additions & 14 deletions

File tree

test/collection/test_classes_generative.py

Lines changed: 13 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -86,7 +86,7 @@ def test_generative_parameters_images_parsing(
8686
GenerativeProvider(
8787
return_metadata=True,
8888
anthropic=GenerativeAnthropic(
89-
base_url="http://localhost:8080/",
89+
base_url="http://localhost:8080",
9090
model="text-to-image",
9191
max_tokens=100,
9292
stop_sequences=TextArray(values=["\n"]),
@@ -107,7 +107,7 @@ def test_generative_parameters_images_parsing(
107107
GenerativeProvider(
108108
return_metadata=True,
109109
anyscale=GenerativeAnyscale(
110-
base_url="http://localhost:8080/",
110+
base_url="http://localhost:8080",
111111
model="text-to-image",
112112
temperature=0.5,
113113
),
@@ -130,7 +130,7 @@ def test_generative_parameters_images_parsing(
130130
GenerativeProvider(
131131
return_metadata=True,
132132
aws=GenerativeAWS(
133-
endpoint="http://localhost:8080/",
133+
endpoint="http://localhost:8080",
134134
model="text-to-image",
135135
region="us-west-2",
136136
service="sagemaker",
@@ -156,7 +156,7 @@ def test_generative_parameters_images_parsing(
156156
GenerativeProvider(
157157
return_metadata=True,
158158
cohere=GenerativeCohere(
159-
base_url="http://localhost:8080/",
159+
base_url="http://localhost:8080",
160160
k=5,
161161
max_tokens=100,
162162
model="text-to-image",
@@ -184,7 +184,7 @@ def test_generative_parameters_images_parsing(
184184
GenerativeProvider(
185185
return_metadata=True,
186186
databricks=GenerativeDatabricks(
187-
endpoint="http://localhost:8080/",
187+
endpoint="http://localhost:8080",
188188
frequency_penalty=0.5,
189189
log_probs=True,
190190
max_tokens=100,
@@ -216,7 +216,7 @@ def test_generative_parameters_images_parsing(
216216
GenerativeProvider(
217217
return_metadata=True,
218218
friendliai=GenerativeFriendliAI(
219-
base_url="http://localhost:8080/",
219+
base_url="http://localhost:8080",
220220
max_tokens=100,
221221
model="text-to-image",
222222
n=5,
@@ -247,7 +247,7 @@ def test_generative_parameters_images_parsing(
247247
GenerativeProvider(
248248
return_metadata=True,
249249
google=GenerativeGoogle(
250-
api_endpoint="localhost:8080/",
250+
api_endpoint="localhost:8080",
251251
endpoint_id="12345678901234567890123456789012",
252252
frequency_penalty=0.5,
253253
max_tokens=100,
@@ -275,7 +275,7 @@ def test_generative_parameters_images_parsing(
275275
GenerativeProvider(
276276
return_metadata=True,
277277
mistral=GenerativeMistral(
278-
base_url="http://localhost:8080/",
278+
base_url="http://localhost:8080",
279279
max_tokens=100,
280280
model="text-to-image",
281281
temperature=0.5,
@@ -294,7 +294,7 @@ def test_generative_parameters_images_parsing(
294294
GenerativeProvider(
295295
return_metadata=True,
296296
nvidia=GenerativeNvidia(
297-
base_url="http://localhost:8080/",
297+
base_url="http://localhost:8080",
298298
max_tokens=100,
299299
model="text-to-image",
300300
temperature=0.5,
@@ -315,7 +315,7 @@ def test_generative_parameters_images_parsing(
315315
GenerativeProvider(
316316
return_metadata=True,
317317
ollama=GenerativeOllama(
318-
api_endpoint="http://localhost:8080/",
318+
api_endpoint="http://localhost:8080",
319319
model="text-to-image",
320320
temperature=0.5,
321321
images=TextArray(values=[LOGO_ENCODED]),
@@ -345,7 +345,7 @@ def test_generative_parameters_images_parsing(
345345
return_metadata=True,
346346
openai=GenerativeOpenAI(
347347
api_version="2020-05-03",
348-
base_url="http://localhost:8080/",
348+
base_url="http://localhost:8080",
349349
deployment_id="12345678901234567890123456789012",
350350
frequency_penalty=0.5,
351351
max_tokens=100,
@@ -383,7 +383,7 @@ def test_generative_parameters_images_parsing(
383383
return_metadata=True,
384384
openai=GenerativeOpenAI(
385385
api_version="2020-05-03",
386-
base_url="http://localhost:8080/",
386+
base_url="http://localhost:8080",
387387
deployment_id="12345678901234567890123456789012",
388388
frequency_penalty=0.5,
389389
max_tokens=100,
@@ -414,7 +414,7 @@ def test_generative_parameters_images_parsing(
414414
GenerativeProvider(
415415
return_metadata=True,
416416
xai=GenerativeXAI(
417-
base_url="http://localhost:8080/",
417+
base_url="http://localhost:8080",
418418
max_tokens=100,
419419
model="text-to-image",
420420
temperature=0.5,

weaviate/collections/classes/generative.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,9 @@
3535

3636

3737
def _parse_anyhttpurl(url: Optional[AnyHttpUrl]) -> Optional[str]:
38-
return str(url) if url is not None else None
38+
if url is None:
39+
return None
40+
return str(url).strip("/")
3941

4042

4143
def _to_text_array(values: Optional[Iterable[str]]) -> Optional[TextArray]:

0 commit comments

Comments
 (0)