uPj %fdZddlmZddlZddlmZddlmZddlm Z m Z ddl m Z Gdd eZ y) au RAGFlow provider configuration for OpenAI-compatible API. RAGFlow provides OpenAI-compatible APIs with unique path structures: - Chat endpoint: /api/v1/chats_openai/{chat_id}/chat/completions - Agent endpoint: /api/v1/agents_openai/{agent_id}/chat/completions Model name format: - Chat: ragflow/chat/{chat_id}/{model_name} - Agent: ragflow/agent/{agent_id}/{model_name} )FinalN)encode_url_path_segment) OpenAIConfig) get_secretget_secret_str)AllMessageValuesceZdZdZdedeeeeffdZ ddedzdedzdeded ed edzdefd Z dededzdedzd edeedzedzeff d Z ddedede e ded ededzdedzdefdZ dede e ded ededef fd ZxZS) RAGFlowConfiga Configuration for RAGFlow OpenAI-compatible API. Handles both chat and agent endpoints by parsing the model name format: - ragflow/chat/{chat_id}/{model_name} for chat endpoints - ragflow/agent/{agent_id}/{model_name} for agent endpoints modelreturnc|jd}t|dkrtd|d|ddk7rtd|d|d}|d vrtd |d |d }dj|d d}|||fS)aa Parse RAGFlow model name format: ragflow/{endpoint_type}/{id}/{model_name} Args: model: Model name in format ragflow/chat/{chat_id}/{model} or ragflow/agent/{agent_id}/{model} Returns: Tuple of (endpoint_type, id, model_name) Raises: ValueError: If model format is invalid /zInvalid RAGFlow model format: zU. Expected format: ragflow/chat/{chat_id}/{model} or ragflow/agent/{agent_id}/{model}rragflowz. Must start with 'ragflow/')chatagentzInvalid RAGFlow endpoint type: z. Must be 'chat' or 'agent'N)splitlen ValueErrorjoin)selfr parts endpoint_type entity_id model_names m/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/llms/ragflow/chat/transformation.py_parse_ragflow_modelz"RAGFlowConfig._parse_ragflow_models{{3' u:>08no  8y =eWD`ab b$Qx  1 1>}oMhij j 8 HHU12Y/ i33Napi_baseapi_keyoptional_paramslitellm_paramsstreamc|r(t|dr|jr|xs |j}|xs*tjxstdxs t d}| t d|j |\}}} |jd}|jdr|dd}n|jdr|dd }t|d } |d k(rd | d} nd| d} | jdsd| z} || S)a Get the complete URL for the RAGFlow API call. Constructs URL based on endpoint type: - Chat: /api/v1/chats_openai/{chat_id}/chat/completions - Agent: /api/v1/agents_openai/{agent_id}/chat/completions Args: api_base: Base API URL (e.g., http://ragflow-server:port or http://ragflow-server:port/v1) api_key: API key (not used in URL construction) model: Model name in format ragflow/{endpoint_type}/{id}/{model} optional_params: Optional parameters litellm_params: LiteLLM parameters (may contain api_base) stream: Whether streaming is enabled Returns: Complete URL for the API call r"RAGFLOW_API_BASENzapi_base is required for RAGFlow provider. Set it via api_base parameter, RAGFLOW_API_BASE environment variable, or litellm.api_baserz/api/v1iz/v1r) field_namerz/api/v1/chats_openai/z/chat/completionsz/api/v1/agents_openai/) hasattrr"litellmrrrr rstripendswithr startswith) rr"r#r r$r%r&rr_encoded_entity_idpaths rget_complete_urlzRAGFlowConfig.get_complete_url?s%8 gnjAnF]F]:>#:#:Hww//w:>P3QwUcdvUw  W  '+&?&?&F# y!??3'   Y '}H   u %}H$;9Q\#] F "*+<*==NOD+,=+>>OPDs#:DD6""r!custom_llm_providerc|j|\}}}|xs*tjxstdxs t d}|xstj xs t d}|||fS)a Get OpenAI-compatible provider information for RAGFlow. Args: model: Model name (will be parsed to extract actual model name) api_base: Base API URL (from input params) api_key: API key (from input params) custom_llm_provider: Custom LLM provider name Returns: Tuple of (api_base, api_key, custom_llm_provider) r(RAGFLOW_API_KEY)r r,r"rrr#) rr r"r#r4r0 actual_modeldynamic_api_basedynamic_api_keys r$_get_openai_compatible_provider_infoz2RAGFlowConfig._get_openai_compatible_provider_infoso*"66u=1l  p(( pJ7I,J pn]oNp  ")!`GOO!`~N_?`2EEEr!headersmessagesc |r(t|dr|jr|xs |j}|xstjxs td}|d||d<d|vr d|vrd|d< |j |\}}} | |d<|S#t $rY|SwxYw) a Validate environment and set up headers for RAGFlow API. Args: headers: Request headers model: Model name messages: Chat messages optional_params: Optional parameters litellm_params: LiteLLM parameters (may contain api_key) api_key: API key (from input params) api_base: Base API URL Returns: Updated headers dictionary r#r6zBearer Authorizationz content-typez Content-Typezapplication/json_ragflow_actual_model)r+r#r,rr r) rr;r r<r$r%r#r"r0r7s rvalidate_environmentz"RAGFlowConfig.validate_environments4 gni@^E[E[7!7!7GQW__Q?P0Q  )0 ':GO $  (^7-J&8GN # !%!:!:5!A Aq,6BN2 3      s$B B  B c|jd}| |j|\}}}t||||||S#t$r|}Y wxYw)a Transform request for RAGFlow API. Uses the actual model name extracted from the RAGFlow model format. Args: model: Model name in RAGFlow format messages: Chat messages optional_params: Optional parameters litellm_params: LiteLLM parameters (may contain _ragflow_actual_model) headers: Request headers Returns: Transformed request dictionary r?)getr rsupertransform_request) rr r<r$r%r;r7r0 __class__s rrDzRAGFlowConfig.transform_requestsj0&))*AB   %%)%>%>u%E"1l w(xR`bijj  %$  %s> A  A )N)NN)__name__ __module__ __qualname____doc__strtupler dictboolr3r:listrr@rD __classcell__)rEs@rr r s4#4%S# 2F4N#>#*>#t># >#  >#  >#t ># >#@FF*Ft F ! F sTz3:s* + FP##111'( 1  1  1t1*1 1f"k"k'("k "k  "k  "k "k"kr!r )rItypingrr,$litellm.litellm_core_utils.url_utilsrlitellm.llms.openai.openairlitellm.secret_managers.mainrrlitellm.types.llms.openairr r!rrVs. H3C6_kL_kr!