109 lines
3.6 KiB
C++
109 lines
3.6 KiB
C++
|
// OpenAI ServiceSample, Copyright LifeEXE. All Rights Reserved.
|
||
|
|
|
||
|
|
#include "Sample/Services/QuestService.h"
|
||
|
|
#include "Funclib/OpenAIFuncLib.h"
|
||
|
|
|
||
|
|
DEFINE_LOG_CATEGORY_STATIC(LogQuestService, All, All);
|
||
|
|
|
||
|
|
FString UQuestService::FunctionName() const
|
||
|
|
{
|
||
|
|
return "get_alien_rampage_saga_characters_info";
|
||
|
|
}
|
||
|
|
|
||
|
|
FString UQuestService::Description() const
|
||
|
|
{
|
||
|
|
return "Get information about characters from my Alien Rampage Saga game.";
|
||
|
|
}
|
||
|
|
|
||
|
|
FString UQuestService::MakeFunction() const
|
||
|
|
{
|
||
|
|
/*
|
||
|
|
"parameters": {
|
||
|
|
"type": "object",
|
||
|
|
"properties": {
|
||
|
|
"name": {
|
||
|
|
"type": "string",
|
||
|
|
"description": "Name of the character of my Alien Rampage Saga.",
|
||
|
|
},
|
||
|
|
"ability": {
|
||
|
|
"type": "boolean",
|
||
|
|
"description": "Set this to true if you want to know what skills the character has.",
|
||
|
|
},
|
||
|
|
},
|
||
|
|
"required": ["name"],
|
||
|
|
}
|
||
|
|
*/
|
||
|
|
|
||
|
|
TSharedPtr<FJsonObject> ParamsObj = MakeShareable(new FJsonObject());
|
||
|
|
ParamsObj->SetStringField("type", "object");
|
||
|
|
|
||
|
|
TSharedPtr<FJsonObject> PropsObj = MakeShareable(new FJsonObject());
|
||
|
|
|
||
|
|
// character name
|
||
|
|
TSharedPtr<FJsonObject> NameObj = MakeShareable(new FJsonObject());
|
||
|
|
NameObj->SetStringField("type", "string");
|
||
|
|
NameObj->SetStringField("description", "Name of the character of my Alien Rampage Saga.");
|
||
|
|
PropsObj->SetObjectField("name", NameObj);
|
||
|
|
|
||
|
|
// ability
|
||
|
|
TSharedPtr<FJsonObject> AbilityObj = MakeShareable(new FJsonObject());
|
||
|
|
AbilityObj->SetStringField("type", "boolean");
|
||
|
|
AbilityObj->SetStringField("description", "Set this to true if you want to know what skills the character has.");
|
||
|
|
PropsObj->SetObjectField("ability", AbilityObj);
|
||
|
|
|
||
|
|
ParamsObj->SetObjectField("properties", PropsObj);
|
||
|
|
|
||
|
|
TArray<TSharedPtr<FJsonValue>> RequiredArray;
|
||
|
|
RequiredArray.Add(MakeShareable(new FJsonValueString("name")));
|
||
|
|
|
||
|
|
ParamsObj->SetArrayField("required", RequiredArray);
|
||
|
|
|
||
|
|
return UOpenAIFuncLib::MakeFunctionsString(ParamsObj);
|
||
|
|
}
|
||
|
|
|
||
|
|
void UQuestService::Call(const TSharedPtr<FJsonObject>& Args, const FString& InToolID)
|
||
|
|
{
|
||
|
|
Super::Call(Args, InToolID);
|
||
|
|
|
||
|
|
FString ArgsStr;
|
||
|
|
if (!UOpenAIFuncLib::JsonToString(Args, ArgsStr))
|
||
|
|
{
|
||
|
|
UE_LOG(LogQuestService, Display, TEXT("Can't convert JSON to string"));
|
||
|
|
}
|
||
|
|
|
||
|
|
FAlienRampageGameProps Props;
|
||
|
|
if (!UOpenAIFuncLib::ParseJSONToStruct<FAlienRampageGameProps>(ArgsStr, &Props))
|
||
|
|
{
|
||
|
|
UE_LOG(LogQuestService, Error, TEXT("Can't parse args"));
|
||
|
|
ServiceDataError.Broadcast("QuestService can't parse args from OpenAI.");
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
|
||
|
|
struct FCharacterInfo
|
||
|
|
{
|
||
|
|
FString Description;
|
||
|
|
FString Abilities;
|
||
|
|
};
|
||
|
|
|
||
|
|
using CharacterName = FString;
|
||
|
|
const TMap<CharacterName, FCharacterInfo> Info //
|
||
|
|
{{"voidwalker", {"The main character fighting for the future of the Universe",
|
||
|
|
"Adept at crafting nasty jokes, has a penchant for donuts, and is mesmerized by the Aurora Borealis"}},
|
||
|
|
{"galaxor", {"A formidable antagonist intent on the Universe's destruction",
|
||
|
|
"Capable of unicorn riding, boasts the ability to sleep for 20 hours straight, and possesses knowledge of all "
|
||
|
|
"credit card PIN codes"}}};
|
||
|
|
|
||
|
|
FString InfoToOpenAI{};
|
||
|
|
const FString Name = Props.Name.ToLower();
|
||
|
|
if (Info.Contains(Name))
|
||
|
|
{
|
||
|
|
InfoToOpenAI = Props.Ability ? Info[Name].Abilities : Info[Name].Description;
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
InfoToOpenAI = "Character with such a name doesn't exist in the game.";
|
||
|
|
}
|
||
|
|
|
||
|
|
ServiceDataRecieved.Broadcast(MakeMessage(InfoToOpenAI));
|
||
|
|
}
|