o ưiu&@s\dZddlmZmZmZddlZddlmZddlm Z m Z ddl m Z GdddeZ dS) 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} )ListOptionalTupleN) OpenAIConfig) get_secretget_secret_str)AllMessageValuescseZdZdZdedeeeeffddZ ddeedeeded ed ed ee defd d Z dedeedeededeeeeeeff ddZ  ddedede e d ed edeedeedefddZdede e d ed ededef fdd ZZS) 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 modelreturncCs|d}t|dkrtd|d|ddkr!td|d|d}|d vr1td |d |d }d|d d}|||fS)a 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}rZragflowz. Must start with 'ragflow/')chatZagentzInvalid RAGFlow endpoint type: z. Must be 'chat' or 'agent'N)splitlen ValueErrorjoin)selfr parts endpoint_type entity_idZ model_namer_/home/app/Keep/.python/lib/python3.10/site-packages/litellm/llms/ragflow/chat/transformation.py_parse_ragflow_models"      z"RAGFlowConfig._parse_ragflow_modelNapi_baseapi_keyoptional_paramslitellm_paramsstreamc Cs|rt|dr|jr|p|j}|ptjptdptd}|dur$td||\}}} |d}|dr=|dd}n |drH|dd }|d krSd |d } nd |d } | dsbd| } || 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 rRAGFLOW_API_BASENzapi_base is required for RAGFlow provider. Set it via api_base parameter, RAGFLOW_API_BASE environment variable, or litellm.api_baser z/api/v1iz/v1rz/api/v1/chats_openai/z/chat/completionsz/api/v1/agents_openai/) hasattrrlitellmrrrrrstripendswith startswith) rrrr rr r!rr_pathrrrget_complete_urlBs.        zRAGFlowConfig.get_complete_urlcustom_llm_providerc CsF||\}}}|ptjptdptd}|ptjptd}|||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)rr%rrrr) rr rrr,r) actual_modelZdynamic_api_baseZdynamic_api_keyrrr$_get_openai_compatible_provider_infos  z2RAGFlowConfig._get_openai_compatible_provider_infoheadersmessagesc Cs|rt|dr|jr|p|j}|ptjptd}|dur#d||d<d|vr/d|vr/d|d<z||\}}} | |d <W|StyHY|Sw) 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 rr-NzBearer Authorizationz content-typez Content-Typezapplication/json_ragflow_actual_model)r$rr%rrr) rr0r r1rr rrr)r.rrrvalidate_environments&   z"RAGFlowConfig.validate_environmentcsR|d}|durz ||\}}}Wn ty|}Ynwt|||||S)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 r3N)getrrsupertransform_request)rr r1rr r0r.r) __class__rrr7s   zRAGFlowConfig.transform_request)N)NN)__name__ __module__ __qualname____doc__strrrrdictboolr+r/rrr4r7 __classcell__rrr8rr s|+ B / 7r )r=typingrrrr%Zlitellm.llms.openai.openairZlitellm.secret_managers.mainrrZlitellm.types.llms.openairr rrrrs