nim_cpp_ai.h

浏览该文件的文档.

 
#ifndef NIM_CPP_AI_H
#define NIM_CPP_AI_H
 
#include <functional>
#include "nim_cpp_wrapper/helper/nim_ai_helper.h"
#include "nim_cpp_wrapper/helper/nim_user_helper.h"
#include "nim_cpp_wrapper/nim_sdk_cpp_wrapper.h"
 
namespace nim {
 
struct NIM_SDK_CPPWRAPPER_DLL_API NIMProxyAIModelCallParams : NIMAIModelCallBase {
    std::string request_id_;
    NIMProxyAICallAntispamConfig antispam_config_;
    void ToJsonValue(nim_cpp_wrapper_util::Json::Value& values) const {
        NIMAIModelCallBase::ToJsonValue(values);
        values[kNIMAICallKeyRequestID] = request_id_;
        antispam_config_.ToJsonValue(values);
    }
    std::string ToJsonString() const {
        nim_cpp_wrapper_util::Json::Value values;
        ToJsonValue(values);
        return GetJsonStringWithNoStyled(values);
    }
};
 
struct NIM_SDK_CPPWRAPPER_DLL_API NIMStopProxyAIModelStreamingCallParams {
    std::string account_id_;
    std::string request_id_;
    void ToJsonValue(nim_cpp_wrapper_util::Json::Value& values) const {
        values[kNIMAICallKeyAccountID] = account_id_;
        values[kNIMAICallKeyRequestID] = request_id_;
    }
    std::string ToJsonString() const {
        nim_cpp_wrapper_util::Json::Value values;
        ToJsonValue(values);
        return GetJsonStringWithNoStyled(values);
    }
};
 
class NIM_SDK_CPPWRAPPER_DLL_API AI {
public:
    using AIModelCallResultCallback = std::function<void(NIMResCode res_code, const NIMAIModelCallResult& result)>;
    using AIModelStreamingCallResultCallback = std::function<void(NIMResCode res_code, const NIMAIModelStreamingCallResult& result)>;
    using GetAIUserListCallback = std::function<void(NIMResCode res_code, const std::list<AIUserNameCard>&)>;
    using ProxyAIModelCallCallback = std::function<void(NIMResCode res_code)>;
    using StopProxyAIModelStreamingCallCallback = ProxyAIModelCallCallback;
    static void RegProxyAIModelCallCb(const AIModelCallResultCallback& callback);
 
    static void RegProxyAIModelStreamingCallCb(const AIModelStreamingCallResultCallback& callback);
 
    static void GetAIUserList(const GetAIUserListCallback& callback);
    static void ProxyAIModelCall(const NIMProxyAIModelCallParams& params, const ProxyAIModelCallCallback& callback);
 
    static void StopProxyAIModelStreamingCall(const NIMStopProxyAIModelStreamingCallParams& params,
        const StopProxyAIModelStreamingCallCallback& callback);
};
 
}  // namespace nim
 
#endif  // NIM_CPP_AI_H