From bef6ddc656f28da680510662ee17ddfa6a43870e Mon Sep 17 00:00:00 2001 From: Stepan Date: Fri, 18 Apr 2025 16:10:03 -0400 Subject: [PATCH] Add base_url parameter to embedding_factor --- src/ragas/embeddings/base.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/src/ragas/embeddings/base.py b/src/ragas/embeddings/base.py index 36338b025..61e2d0684 100644 --- a/src/ragas/embeddings/base.py +++ b/src/ragas/embeddings/base.py @@ -353,7 +353,9 @@ def __repr__(self) -> str: def embedding_factory( - model: str = "text-embedding-ada-002", run_config: t.Optional[RunConfig] = None + model: str = "text-embedding-ada-002", + run_config: t.Optional[RunConfig] = None, + base_url: t.Optional[str] = None, ) -> BaseRagasEmbeddings: """ Create and return a BaseRagasEmbeddings instance. Used for default embeddings @@ -368,13 +370,15 @@ def embedding_factory( The name of the OpenAI embedding model to use, by default "text-embedding-ada-002". run_config : RunConfig, optional Configuration for the run, by default None. + base_url : str, optional + Base URL for the API, by default None. Returns ------- BaseRagasEmbeddings An instance of BaseRagasEmbeddings configured with the specified parameters. """ - openai_embeddings = OpenAIEmbeddings(model=model) + openai_embeddings = OpenAIEmbeddings(model=model, base_url=base_url) if run_config is not None: openai_embeddings.request_timeout = run_config.timeout else: