nemotron3_adapter.h

Go to the documentation of this file.

// SPDX-License-Identifier: Apache-2.0
#pragma once
 
#include <entropic/inference/adapters/adapter_base.h>
 
#include <unordered_map>
 
namespace entropic {
 
class Nemotron3Adapter : public ChatAdapter {
public:
    using ChatAdapter::ChatAdapter;
 
    std::string chat_format() const override { return ""; }
 
    ParseResult parse_tool_calls(const std::string& content) const override;
 
    Message format_tool_result(
        const ToolCall& tool_call,
        const std::string& result) const override;
 
protected:
    std::string format_tools(
        const std::vector<std::string>& tool_jsons) const override;
 
private:
    std::vector<ToolCall> parse_dsml_function_calls(
        const std::string& content) const;
 
    std::unordered_map<std::string, std::string> extract_dsml_parameters(
        const std::string& invoke_body) const;
 
    std::vector<ToolCall> parse_xml_function_calls(
        const std::string& content) const;
 
    std::unordered_map<std::string, std::string> extract_xml_parameters(
        const std::string& func_body) const;
 
    std::string clean_content(const std::string& content) const;
};
 
} // namespace entropic