CompletionTypes.h

// OpenAI, Copyright LifeEXE. All Rights Reserved.
 
#pragma once
 
#include "CoreMinimal.h"
#include "Provider/Types/OpenAICommonTypes.h"
#include "Provider/Types/Chat/ChatTypesHeader.h"
#include "CompletionTypes.generated.h"

//                 REQUEST TYPES
 
USTRUCT(BlueprintType)
struct FCompletion
{
    GENERATED_BODY()
 
    
    UPROPERTY(BlueprintReadWrite, Category = "OpenAI | Required")
    FString Model;

    UPROPERTY(BlueprintReadWrite, Category = "OpenAI | Optional")
    FString Prompt;

    UPROPERTY(BlueprintReadWrite, Category = "OpenAI | Optional")
    FString Suffix;

    UPROPERTY(BlueprintReadWrite, Category = "OpenAI | Optional")
    int32 Max_Tokens{16};

    UPROPERTY(BlueprintReadWrite, Category = "OpenAI | Optional")
    float Temperature{1.0f};

    UPROPERTY(BlueprintReadWrite, Category = "OpenAI | Optional")
    float Top_P{1.0f};

    UPROPERTY(BlueprintReadWrite, Category = "OpenAI | Optional")
    int32 N{1};

    UPROPERTY(BlueprintReadWrite, Category = "OpenAI | Optional")
    bool Stream{false};

    UPROPERTY(BlueprintReadWrite, Category = "OpenAI | Optional")
    int32 Logprobs{0};

    UPROPERTY(BlueprintReadWrite, Category = "OpenAI | Optional")
    bool Echo{false};

    UPROPERTY(BlueprintReadWrite, Category = "OpenAI | Optional")
    float Presence_Penalty{0.0f};

    UPROPERTY(BlueprintReadWrite, Category = "OpenAI | Optional")
    float Frequency_Penalty{0.0f};

    UPROPERTY(BlueprintReadWrite, Category = "OpenAI | Optional")
    int32 Best_Of{1};

    UPROPERTY(BlueprintReadWrite, Category = "OpenAI | Optional")
    TMap<FString, int32> Logit_Bias;

    UPROPERTY(BlueprintReadWrite, Category = "OpenAI | Optional")
    FString User;

    // UPROPERTY(BlueprintReadWrite, Category = "OpenAI | Optional")
    // TArray<FString> Stop;
};

//                 RESPONSE TYPES
 
USTRUCT(BlueprintType)
struct FCompletionLogProbs
{
    GENERATED_BODY()
 
    UPROPERTY(BlueprintReadOnly, Category = "OpenAI")
    TArray<int32> Text_Offset;
 
    UPROPERTY(BlueprintReadOnly, Category = "OpenAI")
    TArray<double> Token_Logprobs;
 
    UPROPERTY(BlueprintReadOnly, Category = "OpenAI")
    TArray<FString> Tokens;
 
    UPROPERTY(BlueprintReadOnly, Category = "OpenAI")
    FString Top_Logprobs{};
};
 
USTRUCT(BlueprintType)
struct FBaseChoice
{
    GENERATED_BODY()
 
    UPROPERTY(BlueprintReadOnly, Category = "OpenAI")
    FString Text{};
 
    UPROPERTY(BlueprintReadOnly, Category = "OpenAI")
    int32 Index{};
};
 
USTRUCT(BlueprintType)
struct FChoice : public FBaseChoice
{
    GENERATED_BODY()
 
    UPROPERTY(BlueprintReadOnly, Category = "OpenAI")
    FCompletionLogProbs LogProbs;

    UPROPERTY(BlueprintReadOnly, Category = "OpenAI")
    FString Finish_Reason{};
};
 
USTRUCT(BlueprintType)
struct FCompletionResponseBase
{
    GENERATED_BODY()
 
    
    UPROPERTY(BlueprintReadOnly, Category = "OpenAI")
    FString ID{};

    UPROPERTY(BlueprintReadOnly, Category = "OpenAI")
    TArray<FChoice> Choices;

    UPROPERTY(BlueprintReadOnly, Category = "OpenAI")
    int32 Created{};

    UPROPERTY(BlueprintReadOnly, Category = "OpenAI")
    FString Model{};

    UPROPERTY(BlueprintReadOnly, Category = "OpenAI")
    FString Object{};
};
 
USTRUCT(BlueprintType)
struct FCompletionResponse : public FCompletionResponseBase
{
    GENERATED_BODY()
 
    
    UPROPERTY(BlueprintReadOnly, Category = "OpenAI")
    FCompletionUsage Usage;
};
 
USTRUCT(BlueprintType)
struct FCompletionStreamResponse : public FCompletionResponseBase
{
    GENERATED_BODY()
};