Skip to content

Model Type Selection with GPT-4 #8

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
wants to merge 3 commits into
base: main
Choose a base branch
from
Open

Model Type Selection with GPT-4 #8

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
4709d5c
Can select Model Type to use in the AI Command Window, added GPT-4 as…
Oscerlot Mar 20, 2023
794c70c
Merge branch 'keijiro:main' into main
Oscerlot Mar 20, 2023
b0a8bb9
Merge branch 'keijiro:main' into main
Oscerlot Mar 21, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
9 changes: 8 additions & 1 deletion Assets/Editor/AICommandWindow.cs
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -35,7 +35,7 @@ static string WrapPrompt(string input)

void RunGenerator()
{
var code = OpenAIUtil.InvokeChat(WrapPrompt(_prompt));
var code = OpenAIUtil.InvokeChat(WrapPrompt(_prompt), _modelType);
Debug.Log("AI command script:" + code);
CreateScriptAsset(code);
}
Expand All @@ -45,6 +45,8 @@ void RunGenerator()
#region Editor GUI

string _prompt = "Create 100 cubes at random points.";

ModelType _modelType = ModelType.gpt_3_5_turbo;

const string ApiKeyErrorText =
"API Key hasn't been set. Please check the project settings " +
Expand All @@ -61,6 +63,11 @@ void OnGUI()
if (IsApiKeyOk)
{
_prompt = EditorGUILayout.TextArea(_prompt, GUILayout.ExpandHeight(true));
_modelType = (ModelType)EditorGUILayout.EnumPopup("Model To Use", _modelType);

if (_modelType == ModelType.gpt_4)
EditorGUILayout.HelpBox("Ensure you have approved API access to GPT-4, or the command will fail!", MessageType.Info);

if (GUILayout.Button("Run")) RunGenerator();
}
else
Expand Down
99 changes: 61 additions & 38 deletions Assets/Editor/OpenAIUtil.cs
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -2,55 +2,78 @@
using UnityEditor;
using UnityEngine.Networking;

namespace AICommand {

static class OpenAIUtil
namespace AICommand
{
static string CreateChatRequestBody(string prompt)
public enum ModelType
{
var msg = new OpenAI.RequestMessage();
msg.role = "user";
msg.content = prompt;

var req = new OpenAI.Request();
req.model = "gpt-3.5-turbo";
req.messages = new [] { msg };
gpt_4,
gpt_3_5_turbo
};

return JsonUtility.ToJson(req);
//Dictionary for the ModelType to the string used by the API
static class ModelTypeDict
{
public static string GetModelTypeString(ModelType modelType)
{
switch (modelType)
{
case ModelType.gpt_4:
return "gpt-4";
default:
Copy link

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Should be a ModelType.gpt_3_5_turbo case before default. More practical for future versions of GPT. I'd also argue the default case should throw an error or at least log that it defaults to 3.5 because the given enum value could not be found.

return "gpt-3.5-turbo";
}
}
}

public static string InvokeChat(string prompt)
static class OpenAIUtil
{
var settings = AICommandSettings.instance;

// POST
using var post = UnityWebRequest.Post
(OpenAI.Api.Url, CreateChatRequestBody(prompt), "application/json");

// Request timeout setting
post.timeout = settings.timeout;
static string CreateChatRequestBody(string prompt, ModelType modelType)
{
var msg = new OpenAI.RequestMessage();
msg.role = "user";
msg.content = prompt;

// API key authorization
post.SetRequestHeader("Authorization", "Bearer " + settings.apiKey);
var req = new OpenAI.Request();
req.model = ModelTypeDict.GetModelTypeString(modelType);
req.messages = new[] { msg };

// Request start
var req = post.SendWebRequest();
return JsonUtility.ToJson(req);
}

// Progress bar (Totally fake! Don't try this at home.)
for (var progress = 0.0f; !req.isDone; progress += 0.01f)
public static string InvokeChat(string prompt, ModelType _modelType)
{
EditorUtility.DisplayProgressBar
("AI Command", "Generating...", progress);
System.Threading.Thread.Sleep(100);
progress += 0.01f;
}
EditorUtility.ClearProgressBar();
var settings = AICommandSettings.instance;

// Response extraction
var json = post.downloadHandler.text;
var data = JsonUtility.FromJson<OpenAI.Response>(json);
return data.choices[0].message.content;
// POST
using var post = UnityWebRequest.Post
(OpenAI.Api.Url, CreateChatRequestBody(prompt, _modelType), "application/json");

// Request timeout setting
post.timeout = settings.timeout;

// API key authorization
post.SetRequestHeader("Authorization", "Bearer " + settings.apiKey);

// Request start
var req = post.SendWebRequest();

// Progress bar (Totally fake! Don't try this at home.)
for (var progress = 0.0f; !req.isDone; progress += 0.01f)
{
EditorUtility.DisplayProgressBar
("AI Command", "Generating...", progress);
System.Threading.Thread.Sleep(100);
progress += 0.01f;
}

EditorUtility.ClearProgressBar();

// Response extraction
var json = post.downloadHandler.text;
var data = JsonUtility.FromJson<OpenAI.Response>(json);
return data.choices[0].message.content;
}
}
}

} // namespace AICommand
// namespace AICommand