Class: Langchain::Assistant::LLM::Adapters::MistralAI
- Defined in:
- lib/langchain/assistant/llm/adapters/mistral_ai.rb
Instance Method Summary collapse
-
#allowed_tool_choices ⇒ Object
Get the allowed assistant.tool_choice values for Mistral AI.
-
#available_tool_names(tools) ⇒ Object
Get the available tool names for Mistral AI.
-
#build_chat_params(messages:, instructions:, tools:, tool_choice:, parallel_tool_calls:) ⇒ Hash
Build the chat parameters for the Mistral AI LLM.
-
#build_message(role:, content: nil, image_url: nil, tool_calls: [], tool_call_id: nil) ⇒ Messages::MistralAIMessage
Build a Mistral AI message.
-
#build_tools(tools) ⇒ Object
Build the tools for the Mistral AI LLM.
-
#extract_tool_call_args(tool_call:) ⇒ Array
Extract the tool call information from the OpenAI tool call hash.
- #support_system_message? ⇒ Boolean
- #tool_role ⇒ Object
Instance Method Details
#allowed_tool_choices ⇒ Object
Get the allowed assistant.tool_choice values for Mistral AI
71 72 73 |
# File 'lib/langchain/assistant/llm/adapters/mistral_ai.rb', line 71 def allowed_tool_choices ["auto", "none"] end |
#available_tool_names(tools) ⇒ Object
Get the available tool names for Mistral AI
76 77 78 |
# File 'lib/langchain/assistant/llm/adapters/mistral_ai.rb', line 76 def available_tool_names(tools) build_tools(tools).map { |tool| tool.dig(:function, :name) } end |
#build_chat_params(messages:, instructions:, tools:, tool_choice:, parallel_tool_calls:) ⇒ Hash
Build the chat parameters for the Mistral AI LLM
16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 |
# File 'lib/langchain/assistant/llm/adapters/mistral_ai.rb', line 16 def build_chat_params( messages:, instructions:, tools:, tool_choice:, parallel_tool_calls: ) Langchain.logger.warn "WARNING: `parallel_tool_calls:` is not supported by Mistral AI currently" params = {messages: } if tools.any? params[:tools] = build_tools(tools) params[:tool_choice] = build_tool_choice(tool_choice) end params end |
#build_message(role:, content: nil, image_url: nil, tool_calls: [], tool_call_id: nil) ⇒ Messages::MistralAIMessage
Build a Mistral AI message
41 42 43 |
# File 'lib/langchain/assistant/llm/adapters/mistral_ai.rb', line 41 def (role:, content: nil, image_url: nil, tool_calls: [], tool_call_id: nil) Messages::MistralAIMessage.new(role: role, content: content, image_url: image_url, tool_calls: tool_calls, tool_call_id: tool_call_id) end |
#build_tools(tools) ⇒ Object
Build the tools for the Mistral AI LLM
66 67 68 |
# File 'lib/langchain/assistant/llm/adapters/mistral_ai.rb', line 66 def build_tools(tools) tools.map { |tool| tool.class.function_schemas.to_openai_format }.flatten end |
#extract_tool_call_args(tool_call:) ⇒ Array
Extract the tool call information from the OpenAI tool call hash
49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 |
# File 'lib/langchain/assistant/llm/adapters/mistral_ai.rb', line 49 def extract_tool_call_args(tool_call:) tool_call_id = tool_call.dig("id") function_name = tool_call.dig("function", "name") tool_name, method_name = function_name.split("__") tool_arguments = tool_call.dig("function", "arguments") tool_arguments = if tool_arguments.is_a?(Hash) Langchain::Utils::HashTransformer.symbolize_keys(tool_arguments) else JSON.parse(tool_arguments, symbolize_names: true) end [tool_call_id, tool_name, method_name, tool_arguments] end |
#support_system_message? ⇒ Boolean
84 85 86 |
# File 'lib/langchain/assistant/llm/adapters/mistral_ai.rb', line 84 def Messages::MistralAIMessage::ROLES.include?("system") end |
#tool_role ⇒ Object
80 81 82 |
# File 'lib/langchain/assistant/llm/adapters/mistral_ai.rb', line 80 def tool_role Messages::MistralAIMessage::TOOL_ROLE end |