CompletionAction.h

// OpenAI, Copyright LifeEXE. All Rights Reserved.
 
#pragma once
 
#include "Kismet/BlueprintAsyncActionBase.h"
#include "Provider/Types/Legacy/CompletionTypes.h"
#include "Provider/Types/CommonTypes.h"
#include "CompletionAction.generated.h"
 
USTRUCT(BlueprintType)
struct FCompletionPayload
{
    GENERATED_BODY()
 
    
    UPROPERTY(BlueprintReadOnly, Category = "OpenAI")
    FCompletionResponse Response;
 
    UPROPERTY(BlueprintReadOnly, Category = "OpenAI")
    TArray<FCompletionStreamResponse> StreamResponse;
 
    UPROPERTY(BlueprintReadOnly, Category = "OpenAI")
    FString StreamResponseString{};
 
    UPROPERTY(BlueprintReadOnly, Category = "OpenAI")
    bool bStream{};
 
    UPROPERTY(BlueprintReadOnly, Category = "OpenAI")
    bool bCompleted{};
};
 
DECLARE_DYNAMIC_MULTICAST_DELEGATE_ThreeParams(
    FOnCompletion, const FCompletionPayload&, Payload, const FOpenAIResponseMetadata&, ResponseMetadata, const FOpenAIError&, RawError);
 
class UOpenAIProvider;
 
UCLASS()
class UCompletionAction : public UBlueprintAsyncActionBase
{
    GENERATED_BODY()
 
public:
    UPROPERTY(BlueprintAssignable)
    FOnCompletion OnUpdate;
 
    virtual void Activate() override;
 
private:
    UFUNCTION(BlueprintCallable, meta = (BlueprintInternalUseOnly = "true"), Category = "OpenAI | Completion")
    static UCompletionAction* CreateCompletion(const FCompletion& CompletionRequest, const FOpenAIAuth& Auth, const FString& URLOverride);
 
    void TryToOverrideURL(UOpenAIProvider* Provider);
 
    void OnCreateCompletionCompleted(const FCompletionResponse& Response, const FOpenAIResponseMetadata& ResponseMetadata);
    void OnCreateCompletionStreamProgresses(
        const TArray<FCompletionStreamResponse>& Responses, const FOpenAIResponseMetadata& ResponseMetadata);
    void OnCreateCompletionStreamCompleted(
        const TArray<FCompletionStreamResponse>& Responses, const FOpenAIResponseMetadata& ResponseMetadata);
    void OnRequestError(const FString& URL, const FString& Content);
 
private:
    FCompletion Completion;
    FOpenAIAuth Auth;
    FString URLOverride{};
};