|
55 | 55 | ) |
56 | 56 | from weaviate.exceptions import ( |
57 | 57 | WeaviateInvalidInputError, |
58 | | - WeaviateUnsupportedFeatureError, |
59 | 58 | ) |
60 | 59 | from weaviate.proto.v1 import generative_pb2, search_get_pb2 |
61 | 60 | from weaviate.types import INCLUDE_VECTOR, UUID, UUIDS |
@@ -304,96 +303,74 @@ def __init__( |
304 | 303 | self.generative_provider = generative_provider |
305 | 304 |
|
306 | 305 | def to_grpc(self, server_version: _ServerVersion) -> generative_pb2.GenerativeSearch: |
307 | | - if server_version.is_lower_than(1, 27, 14): |
308 | | - if self.generative_provider is not None: |
309 | | - raise WeaviateUnsupportedFeatureError("Dynamic RAG", str(server_version), "1.30.0") |
310 | | - |
311 | | - if isinstance(self.single, _SinglePrompt): |
312 | | - single_prompt: Optional[str] = self.single.prompt |
313 | | - else: |
314 | | - single_prompt = self.single |
315 | | - |
316 | | - if isinstance(self.grouped, _GroupedTask): |
317 | | - grouped_task: Optional[str] = self.grouped.prompt |
318 | | - grouped_properties = self.grouped.non_blob_properties |
319 | | - else: |
320 | | - grouped_task = self.grouped |
321 | | - grouped_properties = self.grouped_properties |
322 | | - |
323 | | - return generative_pb2.GenerativeSearch( |
324 | | - single_response_prompt=single_prompt, |
325 | | - grouped_response_task=grouped_task, |
326 | | - grouped_properties=grouped_properties, |
327 | | - ) |
328 | | - else: |
329 | | - single: Optional[generative_pb2.GenerativeSearch.Single] = None |
330 | | - if isinstance(self.single, _SinglePrompt): |
331 | | - single = generative_pb2.GenerativeSearch.Single( |
332 | | - prompt=self.single.prompt, |
333 | | - queries=( |
334 | | - [ |
335 | | - self.generative_provider._to_grpc( |
336 | | - _GenerativeConfigRuntimeOptions( |
337 | | - self.single.metadata, |
338 | | - self.single.images, |
339 | | - self.single.image_properties, |
340 | | - ) |
| 306 | + single: Optional[generative_pb2.GenerativeSearch.Single] = None |
| 307 | + if isinstance(self.single, _SinglePrompt): |
| 308 | + single = generative_pb2.GenerativeSearch.Single( |
| 309 | + prompt=self.single.prompt, |
| 310 | + queries=( |
| 311 | + [ |
| 312 | + self.generative_provider._to_grpc( |
| 313 | + _GenerativeConfigRuntimeOptions( |
| 314 | + self.single.metadata, |
| 315 | + self.single.images, |
| 316 | + self.single.image_properties, |
341 | 317 | ) |
342 | | - ] |
343 | | - if self.generative_provider is not None |
344 | | - else None |
345 | | - ), |
346 | | - debug=self.single.debug, |
347 | | - ) |
348 | | - if isinstance(self.single, str): |
349 | | - single = generative_pb2.GenerativeSearch.Single( |
350 | | - prompt=self.single, |
351 | | - queries=( |
352 | | - [self.generative_provider._to_grpc(_GenerativeConfigRuntimeOptions())] |
353 | | - if self.generative_provider is not None |
354 | | - else None |
355 | | - ), |
356 | | - ) |
357 | | - |
358 | | - grouped: Optional[generative_pb2.GenerativeSearch.Grouped] = None |
359 | | - if isinstance(self.grouped, _GroupedTask): |
360 | | - grouped = generative_pb2.GenerativeSearch.Grouped( |
361 | | - task=self.grouped.prompt, |
362 | | - properties=_to_text_array(self.grouped.non_blob_properties), |
363 | | - queries=( |
364 | | - [ |
365 | | - self.generative_provider._to_grpc( |
366 | | - _GenerativeConfigRuntimeOptions( |
367 | | - self.grouped.metadata, |
368 | | - self.grouped.images, |
369 | | - self.grouped.image_properties, |
370 | | - ) |
| 318 | + ) |
| 319 | + ] |
| 320 | + if self.generative_provider is not None |
| 321 | + else None |
| 322 | + ), |
| 323 | + debug=self.single.debug, |
| 324 | + ) |
| 325 | + if isinstance(self.single, str): |
| 326 | + single = generative_pb2.GenerativeSearch.Single( |
| 327 | + prompt=self.single, |
| 328 | + queries=( |
| 329 | + [self.generative_provider._to_grpc(_GenerativeConfigRuntimeOptions())] |
| 330 | + if self.generative_provider is not None |
| 331 | + else None |
| 332 | + ), |
| 333 | + ) |
| 334 | + |
| 335 | + grouped: Optional[generative_pb2.GenerativeSearch.Grouped] = None |
| 336 | + if isinstance(self.grouped, _GroupedTask): |
| 337 | + grouped = generative_pb2.GenerativeSearch.Grouped( |
| 338 | + task=self.grouped.prompt, |
| 339 | + properties=_to_text_array(self.grouped.non_blob_properties), |
| 340 | + queries=( |
| 341 | + [ |
| 342 | + self.generative_provider._to_grpc( |
| 343 | + _GenerativeConfigRuntimeOptions( |
| 344 | + self.grouped.metadata, |
| 345 | + self.grouped.images, |
| 346 | + self.grouped.image_properties, |
371 | 347 | ) |
372 | | - ] |
373 | | - if self.generative_provider is not None |
374 | | - else None |
375 | | - ), |
376 | | - ) |
377 | | - if isinstance(self.grouped, str): |
378 | | - grouped = generative_pb2.GenerativeSearch.Grouped( |
379 | | - task=self.grouped, |
380 | | - properties=( |
381 | | - _to_text_array(self.grouped_properties) |
382 | | - if self.grouped_properties is not None |
383 | | - else None |
384 | | - ), |
385 | | - queries=( |
386 | | - [self.generative_provider._to_grpc(_GenerativeConfigRuntimeOptions())] |
387 | | - if self.generative_provider is not None |
388 | | - else None |
389 | | - ), |
390 | | - ) |
391 | | - |
392 | | - return generative_pb2.GenerativeSearch( |
393 | | - single=single, |
394 | | - grouped=grouped, |
| 348 | + ) |
| 349 | + ] |
| 350 | + if self.generative_provider is not None |
| 351 | + else None |
| 352 | + ), |
| 353 | + ) |
| 354 | + if isinstance(self.grouped, str): |
| 355 | + grouped = generative_pb2.GenerativeSearch.Grouped( |
| 356 | + task=self.grouped, |
| 357 | + properties=( |
| 358 | + _to_text_array(self.grouped_properties) |
| 359 | + if self.grouped_properties is not None |
| 360 | + else None |
| 361 | + ), |
| 362 | + queries=( |
| 363 | + [self.generative_provider._to_grpc(_GenerativeConfigRuntimeOptions())] |
| 364 | + if self.generative_provider is not None |
| 365 | + else None |
| 366 | + ), |
395 | 367 | ) |
396 | 368 |
|
| 369 | + return generative_pb2.GenerativeSearch( |
| 370 | + single=single, |
| 371 | + grouped=grouped, |
| 372 | + ) |
| 373 | + |
397 | 374 |
|
398 | 375 | class _GroupBy: |
399 | 376 | prop: str |
|
0 commit comments