Feat: beelzebub cloud integrations (#117)

* improve beelzebub cloud integration

* refactoring cloud integration, fix unit test

* add unit test get honeypots

* improve code coverage
This commit is contained in:
Mario Candela
2024-08-01 20:05:05 +02:00
committed by GitHub
parent cd284877cf
commit a1e96738fb
12 changed files with 276 additions and 50 deletions

View File

@ -285,3 +285,16 @@ func TestBuildExecuteModelHTTPWithoutResults(t *testing.T) {
//Then
assert.Equal(t, "no choices", err.Error())
}
func TestFromString(t *testing.T) {
model, err := FromStringToLLMModel("llama3")
assert.Nil(t, err)
assert.Equal(t, LLAMA3, model)
model, err = FromStringToLLMModel("gpt4-o")
assert.Nil(t, err)
assert.Equal(t, GPT4O, model)
model, err = FromStringToLLMModel("beelzebub-model")
assert.Errorf(t, err, "model beelzebub-model not found")
}