VaalaCat/go-openai
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix for broken Azure Assistants url (#665)

Browse Source 
* fix:fix url for Azure assistants api

* test:add unit tests for Azure Assistants api

* fix:minor liniting issue
This commit is contained in:
Rich Coggins
2024-02-21 07:45:15 -05:00
committed by GitHub
parent e8b347891b
commit 7381d18a75
2 changed files with 191 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

190
assistant_test.go
View File

@@ -202,3 +202,193 @@ When asked a question, write and run Python code to answer the question.`
err = client.DeleteAssistantFile(ctx, assistantID, assistantFileID)
checks.NoError(t, err, "DeleteAssistantFile error")
}
func TestAzureAssistant(t *testing.T) {
assistantID := "asst_abc123"
assistantName := "Ambrogio"
assistantDescription := "Ambrogio is a friendly assistant."
assistantInstructions := `You are a personal math tutor.
When asked a question, write and run Python code to answer the question.`
assistantFileID := "file-wB6RM6wHdA49HfS2DJ9fEyrH"
limit := 20
order := "desc"
after := "asst_abc122"
before := "asst_abc124"
client, server, teardown := setupAzureTestServer()
defer teardown()
server.RegisterHandler(
"/openai/assistants/"+assistantID+"/files/"+assistantFileID,
func(w http.ResponseWriter, r *http.Request) {
if r.Method == http.MethodGet {
resBytes, _ := json.Marshal(openai.AssistantFile{
ID: assistantFileID,
Object: "assistant.file",
CreatedAt: 1234567890,
AssistantID: assistantID,
})
fmt.Fprintln(w, string(resBytes))
} else if r.Method == http.MethodDelete {
fmt.Fprintln(w, `{
id: "file-wB6RM6wHdA49HfS2DJ9fEyrH",
object: "assistant.file.deleted",
deleted: true
}`)
}
},
)
server.RegisterHandler(
"/openai/assistants/"+assistantID+"/files",
func(w http.ResponseWriter, r *http.Request) {
if r.Method == http.MethodGet {
resBytes, _ := json.Marshal(openai.AssistantFilesList{
AssistantFiles: []openai.AssistantFile{
{
ID: assistantFileID,
Object: "assistant.file",
CreatedAt: 1234567890,
AssistantID: assistantID,
},
},
})
fmt.Fprintln(w, string(resBytes))
} else if r.Method == http.MethodPost {
var request openai.AssistantFileRequest
err := json.NewDecoder(r.Body).Decode(&request)
checks.NoError(t, err, "Decode error")
resBytes, _ := json.Marshal(openai.AssistantFile{
ID: request.FileID,
Object: "assistant.file",
CreatedAt: 1234567890,
AssistantID: assistantID,
})
fmt.Fprintln(w, string(resBytes))
}
},
)
server.RegisterHandler(
"/openai/assistants/"+assistantID,
func(w http.ResponseWriter, r *http.Request) {
switch r.Method {
case http.MethodGet:
resBytes, _ := json.Marshal(openai.Assistant{
ID: assistantID,
Object: "assistant",
CreatedAt: 1234567890,
Name: &assistantName,
Model: openai.GPT4TurboPreview,
Description: &assistantDescription,
Instructions: &assistantInstructions,
})
fmt.Fprintln(w, string(resBytes))
case http.MethodPost:
var request openai.AssistantRequest
err := json.NewDecoder(r.Body).Decode(&request)
checks.NoError(t, err, "Decode error")
resBytes, _ := json.Marshal(openai.Assistant{
ID: assistantID,
Object: "assistant",
CreatedAt: 1234567890,
Name: request.Name,
Model: request.Model,
Description: request.Description,
Instructions: request.Instructions,
Tools: request.Tools,
})
fmt.Fprintln(w, string(resBytes))
case http.MethodDelete:
fmt.Fprintln(w, `{
"id": "asst_abc123",
"object": "assistant.deleted",
"deleted": true
}`)
}
},
)
server.RegisterHandler(
"/openai/assistants",
func(w http.ResponseWriter, r *http.Request) {
if r.Method == http.MethodPost {
var request openai.AssistantRequest
err := json.NewDecoder(r.Body).Decode(&request)
checks.NoError(t, err, "Decode error")
resBytes, _ := json.Marshal(openai.Assistant{
ID: assistantID,
Object: "assistant",
CreatedAt: 1234567890,
Name: request.Name,
Model: request.Model,
Description: request.Description,
Instructions: request.Instructions,
Tools: request.Tools,
})
fmt.Fprintln(w, string(resBytes))
} else if r.Method == http.MethodGet {
resBytes, _ := json.Marshal(openai.AssistantsList{
LastID: &assistantID,
FirstID: &assistantID,
Assistants: []openai.Assistant{
{
ID: assistantID,
Object: "assistant",
CreatedAt: 1234567890,
Name: &assistantName,
Model: openai.GPT4TurboPreview,
Description: &assistantDescription,
Instructions: &assistantInstructions,
},
},
})
fmt.Fprintln(w, string(resBytes))
}
},
)
ctx := context.Background()
_, err := client.CreateAssistant(ctx, openai.AssistantRequest{
Name: &assistantName,
Description: &assistantDescription,
Model: openai.GPT4TurboPreview,
Instructions: &assistantInstructions,
})
checks.NoError(t, err, "CreateAssistant error")
_, err = client.RetrieveAssistant(ctx, assistantID)
checks.NoError(t, err, "RetrieveAssistant error")
_, err = client.ModifyAssistant(ctx, assistantID, openai.AssistantRequest{
Name: &assistantName,
Description: &assistantDescription,
Model: openai.GPT4TurboPreview,
Instructions: &assistantInstructions,
})
checks.NoError(t, err, "ModifyAssistant error")
_, err = client.DeleteAssistant(ctx, assistantID)
checks.NoError(t, err, "DeleteAssistant error")
_, err = client.ListAssistants(ctx, &limit, &order, &after, &before)
checks.NoError(t, err, "ListAssistants error")
_, err = client.CreateAssistantFile(ctx, assistantID, openai.AssistantFileRequest{
FileID: assistantFileID,
})
checks.NoError(t, err, "CreateAssistantFile error")
_, err = client.ListAssistantFiles(ctx, assistantID, &limit, &order, &after, &before)
checks.NoError(t, err, "ListAssistantFiles error")
_, err = client.RetrieveAssistantFile(ctx, assistantID, assistantFileID)
checks.NoError(t, err, "RetrieveAssistantFile error")
err = client.DeleteAssistantFile(ctx, assistantID, assistantFileID)
checks.NoError(t, err, "DeleteAssistantFile error")
}

2
client.go
View File

@@ -221,7 +221,7 @@ func (c *Client) fullURL(suffix string, args ...any) string {
baseURL = strings.TrimRight(baseURL, "/")
// if suffix is /models change to {endpoint}/openai/models?api-version=2022-12-01
// https://learn.microsoft.com/en-us/rest/api/cognitiveservices/azureopenaistable/models/list?tabs=HTTP
if strings.Contains(suffix, "/models") {
if strings.Contains(suffix, "/models") || strings.Contains(suffix, "/assistants") {
return fmt.Sprintf("%s/%s%s?api-version=%s", baseURL, azureAPIPrefix, suffix, c.config.APIVersion)
}
azureDeploymentName := "UNKNOWN"