forked from undreamai/LLMUnity
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathServerClient.cs
More file actions
81 lines (69 loc) · 1.99 KB
/
Copy pathServerClient.cs
File metadata and controls
81 lines (69 loc) · 1.99 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
using UnityEngine;
using LLMUnity;
using UnityEngine.UI;
namespace LLMUnitySamples
{
public class ServerClientInteraction
{
InputField playerText;
Text AIText;
LLMClient llm;
public ServerClientInteraction(InputField playerText, Text AIText, LLMClient llm)
{
this.playerText = playerText;
this.AIText = AIText;
this.llm = llm;
}
public void Start()
{
playerText.onSubmit.AddListener(onInputFieldSubmit);
playerText.Select();
}
public void onInputFieldSubmit(string message)
{
playerText.interactable = false;
AIText.text = "...";
_ = llm.Chat(message, SetAIText, AIReplyComplete);
}
public void SetAIText(string text)
{
AIText.text = text;
}
public void AIReplyComplete()
{
playerText.interactable = true;
playerText.Select();
playerText.text = "";
}
}
public class ServerClient : MonoBehaviour
{
public LLM llm;
public InputField playerText1;
public Text AIText1;
ServerClientInteraction interaction1;
public LLMClient llmClient;
public InputField playerText2;
public Text AIText2;
ServerClientInteraction interaction2;
void Start()
{
interaction1 = new ServerClientInteraction(playerText1, AIText1, llm);
interaction2 = new ServerClientInteraction(playerText2, AIText2, llmClient);
interaction1.Start();
interaction2.Start();
}
public void CancelRequests()
{
llm.CancelRequests();
llmClient.CancelRequests();
interaction1.AIReplyComplete();
interaction2.AIReplyComplete();
}
public void ExitGame()
{
Debug.Log("Exit button clicked");
Application.Quit();
}
}
}