mirror of
https://github.com/mariocandela/beelzebub.git
synced 2025-07-01 18:47:26 -04:00
b1de020de8
* Change OpenAI model and prompt. --------- Co-authored-by: mariocandela <mario.candela.personal@gmail.com>
117 lines
3.2 KiB
Go
117 lines
3.2 KiB
Go
package plugins
|
|
|
|
import (
|
|
"github.com/go-resty/resty/v2"
|
|
"github.com/jarcoal/httpmock"
|
|
"github.com/stretchr/testify/assert"
|
|
"net/http"
|
|
"testing"
|
|
)
|
|
|
|
func TestBuildPromptEmptyHistory(t *testing.T) {
|
|
//Given
|
|
var histories []History
|
|
command := "pwd"
|
|
|
|
//When
|
|
prompt := buildPrompt(histories, command)
|
|
|
|
//Then
|
|
assert.Equal(t,
|
|
"You will act as an Ubuntu Linux terminal. The user will type commands, and you are to reply with what the terminal should show. Your responses must be contained within a single code block. Do not provide explanations or type commands unless explicitly instructed by the user. Remember previous commands and consider their effects on subsequent outputs.\n\nA:pwd\n\nQ:/home/user\n\nA:pwd\n\nQ:",
|
|
prompt)
|
|
}
|
|
|
|
func TestBuildPromptWithHistory(t *testing.T) {
|
|
//Given
|
|
var histories = []History{
|
|
{
|
|
Input: "cat hello.txt",
|
|
Output: "world",
|
|
},
|
|
{
|
|
Input: "echo 1234",
|
|
Output: "1234",
|
|
},
|
|
}
|
|
|
|
command := "pwd"
|
|
|
|
//When
|
|
prompt := buildPrompt(histories, command)
|
|
|
|
//Then
|
|
assert.Equal(t,
|
|
"You will act as an Ubuntu Linux terminal. The user will type commands, and you are to reply with what the terminal should show. Your responses must be contained within a single code block. Do not provide explanations or type commands unless explicitly instructed by the user. Remember previous commands and consider their effects on subsequent outputs.\n\nA:pwd\n\nQ:/home/user\n\nA:cat hello.txt\n\nQ:world\n\nA:echo 1234\n\nQ:1234\n\nA:pwd\n\nQ:",
|
|
prompt)
|
|
}
|
|
|
|
func TestBuildGetCompletionsFailValidation(t *testing.T) {
|
|
openAIGPTVirtualTerminal := Init(make([]History, 0), "")
|
|
|
|
_, err := openAIGPTVirtualTerminal.GetCompletions("test")
|
|
|
|
assert.Equal(t, "openAIKey is empty", err.Error())
|
|
}
|
|
|
|
func TestBuildGetCompletionsWithResults(t *testing.T) {
|
|
client := resty.New()
|
|
httpmock.ActivateNonDefault(client.GetClient())
|
|
defer httpmock.DeactivateAndReset()
|
|
|
|
// Given
|
|
httpmock.RegisterResponder("POST", openAIGPTEndpoint,
|
|
func(req *http.Request) (*http.Response, error) {
|
|
resp, err := httpmock.NewJsonResponse(200, &gptResponse{
|
|
Choices: []Choice{
|
|
{
|
|
Text: "prova.txt",
|
|
},
|
|
},
|
|
})
|
|
if err != nil {
|
|
return httpmock.NewStringResponse(500, ""), nil
|
|
}
|
|
return resp, nil
|
|
},
|
|
)
|
|
|
|
openAIGPTVirtualTerminal := Init(make([]History, 0), "sdjdnklfjndslkjanfk")
|
|
openAIGPTVirtualTerminal.client = client
|
|
|
|
//When
|
|
str, err := openAIGPTVirtualTerminal.GetCompletions("ls")
|
|
|
|
//Then
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, "prova.txt", str)
|
|
}
|
|
|
|
func TestBuildGetCompletionsWithoutResults(t *testing.T) {
|
|
client := resty.New()
|
|
httpmock.ActivateNonDefault(client.GetClient())
|
|
defer httpmock.DeactivateAndReset()
|
|
|
|
// Given
|
|
httpmock.RegisterResponder("POST", openAIGPTEndpoint,
|
|
func(req *http.Request) (*http.Response, error) {
|
|
resp, err := httpmock.NewJsonResponse(200, &gptResponse{
|
|
Choices: []Choice{},
|
|
})
|
|
if err != nil {
|
|
return httpmock.NewStringResponse(500, ""), nil
|
|
}
|
|
return resp, nil
|
|
},
|
|
)
|
|
|
|
openAIGPTVirtualTerminal := Init(make([]History, 0), "sdjdnklfjndslkjanfk")
|
|
openAIGPTVirtualTerminal.client = client
|
|
|
|
//When
|
|
_, err := openAIGPTVirtualTerminal.GetCompletions("ls")
|
|
|
|
//Then
|
|
assert.Equal(t, "no choices", err.Error())
|
|
}
|