ChatCompletionAction.h

// OpenAI, Copyright LifeEXE. All Rights Reserved.
 
#pragma once
 
#include "BlueprintAsyncActions/OpenAIActionBase.h"
#include "Provider/Types/Chat/ChatCompletionTypes.h"
#include "Provider/Types/Chat/ChatCompletionChunkTypes.h"
#include "Provider/Types/OpenAICommonTypes.h"
#include "ChatCompletionAction.generated.h"
 
USTRUCT(BlueprintType)
struct FChatCompletionPayload
{
    GENERATED_BODY()
 
    
    UPROPERTY(BlueprintReadOnly, Category = "OpenAI")
    FChatCompletionResponse Response;

    UPROPERTY(BlueprintReadOnly, Category = "OpenAI")
    TArray<FChatCompletionStreamResponse> StreamResponse;

    UPROPERTY(BlueprintReadOnly, Category = "OpenAI")
    FString StreamResponseString{};
 
    UPROPERTY(BlueprintReadOnly, Category = "OpenAI")
    bool bStream{};
 
    UPROPERTY(BlueprintReadOnly, Category = "OpenAI")
    bool bCompleted{};
};
 
DECLARE_DYNAMIC_MULTICAST_DELEGATE_ThreeParams(FOnChatCompletion, const FChatCompletionPayload&, Payload, const FOpenAIResponseMetadata&,
    ResponseMetadata, const FOpenAIError&, RawError);
 
UCLASS()
class OPENAI_API UChatCompletionAction : public UOpenAIActionBase
{
    GENERATED_BODY()
 
public:
    UPROPERTY(BlueprintAssignable)
    FOnChatCompletion OnUpdate;
 
    virtual void Activate() override;
 
private:
    UFUNCTION(BlueprintCallable, meta = (BlueprintInternalUseOnly = "true"), Category = "OpenAI | Chat")
    static UChatCompletionAction* CreateChatCompletion(
        const FChatCompletion& ChatCompletion, const FOpenAIAuth& Auth, const FString& URLOverride);
 
    void OnCreateChatCompletionCompleted(const FChatCompletionResponse& Response, const FOpenAIResponseMetadata& ResponseMetadata);
    void OnCreateChatCompletionStreamProgresses(
        const TArray<FChatCompletionStreamResponse>& Responses, const FOpenAIResponseMetadata& ResponseMetadata);
    void OnCreateChatCompletionStreamCompleted(
        const TArray<FChatCompletionStreamResponse>& Responses, const FOpenAIResponseMetadata& ResponseMetadata);
    virtual void OnRequestError(const FString& URL, const FString& Content) override;
    virtual void SetEndpoint(OpenAI::V1::FOpenAIEndpoints& Endpoints, const FString& URL) const override;
 
private:
    FChatCompletion ChatCompletion;
};