WeatherService.h

// OpenAI, Copyright LifeEXE. All Rights Reserved.
 
#pragma once
 
#include "CoreMinimal.h"
#include "ChatGPT/BaseService.h"
#include "HTTP.h"
#include "WeatherService.generated.h"
 
USTRUCT()
struct FWeatherLocation
{
    GENERATED_BODY()
 
    UPROPERTY()
    FString Name;  // city
 
    UPROPERTY()
    FString Country;
};
 
USTRUCT()
struct FCurrentWeather
{
    GENERATED_BODY()
 
    UPROPERTY()
    FString Name;
 
    UPROPERTY()
    double Temperature{0.0};
 
    UPROPERTY()
    TArray<FString> Weather_Descriptions;
};
 
USTRUCT()
struct FWeather
{
    GENERATED_BODY()
 
    UPROPERTY()
    FWeatherLocation Location;
 
    UPROPERTY()
    FCurrentWeather Current;
 
    // @todo: more fields can be added
    // https://weatherstack.com/documentation
};
 
UCLASS()
class OPENAI_API UWeatherService : public UBaseService
{
    GENERATED_BODY()
 
public:
    virtual bool Init(const OpenAI::ServiceSecrets& Secrets) override;
    virtual FString Name() const override { return "Weather"; }
    virtual FString TooltipDescription() const override { return "Weather from https://weatherstack.com"; }
    virtual FString Description() const override;
    virtual FString FunctionName() const override;
    virtual void Call(const TSharedPtr<FJsonObject>& Args, const FString& ToolID) override;
 
protected:
    virtual FString MakeFunction() const;
 
private:
    FString API_KEY{};
 
    bool MakeRequestURL(const TSharedPtr<FJsonObject>& ArgsJson, FString& WeatherRequestURL) const;
    void OnRequestCompleted(FHttpRequestPtr Request, FHttpResponsePtr Response, bool WasSuccessful);
    void SendError(const FString& ErrorMessage);
};