OpenAI.hpp

浏览该文件的文档.

 
 
#pragma once
 
#include "AstGlobal.h"
#include "AstAI/LLMClient.hpp"
#include "AstAI/ChatMessage.hpp"
#include "AstAI/ChatTool.hpp"
#include "AstUtil/StringView.hpp"
#include <string>
#include <vector>
#include <map>
 
AST_NAMESPACE_BEGIN
 
class JsonValue;
 
class AST_AI_API OpenAI: public LLMClient {
public:
    OpenAI();
 
    OpenAI(const std::string& apiKey, const std::string& baseUrl);
 
    #if 0
    std::string chat(
        const std::string& model, const std::vector<ChatMessage>& messages, 
        const std::vector<ChatTool>& tools = {}, float temperature = 0.7f
    );
    #endif
public: // 底层json接口
 
    using LLMClient::chat;  // 引入基类便捷封装 chat(JsonValue)
 
    errc_t chat(const JsonValue& request, JsonValue& response) override;
 
    errc_t chatStream(const JsonValue& request,
                      ChatEventHandler& handler,
                      JsonValue& accumulatedResult) override;
 
    void setApiKey(StringView apiKey){apiKey_ = std::string(apiKey);}
 
    void setBaseUrl(StringView baseUrl){baseUrl_ = std::string(baseUrl);}
protected:
    const std::string& apiKey() const { return apiKey_; }
    const std::string& baseUrl() const { return baseUrl_; }
private:
    std::string apiKey_;          
    std::string baseUrl_;         
};
 
AST_NAMESPACE_END