mirror of
https://github.com/BerriAI/litellm.git
synced 2026-10-09 03:18:44 +00:00
style: format s3 vectors transformation and rag endpoints
This commit is contained in:
parent
6012f893fa
commit
cf4738c3b7
2 changed files with 7 additions and 2 deletions
|
|
@ -68,7 +68,9 @@ class S3VectorsVectorStoreConfig(BaseVectorStoreConfig, BaseAWSLLM):
|
|||
"""Return the router iff it serves ``embedding_model`` as a deployment."""
|
||||
if router is None:
|
||||
return None
|
||||
model_list: Final = [dict(m) for m in (router.get_model_list() or ())] # mutable-ok: resolve_embedding_router requires list[dict]
|
||||
model_list: Final = [
|
||||
dict(m) for m in (router.get_model_list() or ())
|
||||
] # mutable-ok: resolve_embedding_router requires list[dict]
|
||||
return resolve_embedding_router(embedding_model=embedding_model, llm_router=router, llm_model_list=model_list)
|
||||
|
||||
def transform_search_vector_store_request(
|
||||
|
|
|
|||
|
|
@ -717,7 +717,10 @@ async def rag_query(
|
|||
vector_store_id=retrieval_config["vector_store_id"],
|
||||
user_api_key_dict=user_api_key_dict,
|
||||
)
|
||||
merged_retrieval_config: Final = {**store_data, **retrieval_config} # mutable-ok: litellm.aquery requires a plain dict payload
|
||||
merged_retrieval_config: Final = {
|
||||
**store_data,
|
||||
**retrieval_config,
|
||||
} # mutable-ok: litellm.aquery requires a plain dict payload
|
||||
|
||||
# Add litellm data
|
||||
request_data: dict[str, object] = {}
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue