CompletionTypes.h

// OpenAI, Copyright LifeEXE. All Rights Reserved.
 
#pragma once
 
#include "CoreMinimal.h"
#include "Provider/Types/CommonTypes.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()
};