[azopenaiassistants] Updates for 2024-10-01-preview (Azure#23730)

* hash+gen

* Adding in MessageAttachmentToolDefinition

* Accidental copy/paste

* Updating doc comment for MessageAttachmentToolDefinition

* Update docindex.yml to use Python 3.11 (Azure#23719)

Co-authored-by: Daniel Jurek <[email protected]>

* Refresh OIDC token in azadmin test resources scripts (Azure#23722)

* remove pkg.go.dev link in readme (Azure#23708)

* remove pkg.go.dev link in readme

* fix changelog

* upgrade typesspec-go (Azure#23720)

* [azopenai] Inference changes for 2024-10-01-preview (Azure#23672)

* update for 1.2.0 release (Azure#23728)

* Sync eng/common directory with azure-sdk-tools for PR 9328 (Azure#23692)

* ensure that ChangedServices value within the generated pr diff is always properly an array of strings
* upload the changed services as an attachment named AdditionalTags in save-package-properties.yml, to allow PipelineWitness to tag public builds during log processing

---------

Co-authored-by: Scott Beddall <[email protected]>
Co-authored-by: Scott Beddall <[email protected]>
Co-authored-by: Patrick Hallisey <[email protected]>

* refine typespec release tool (Azure#23735)

* Added permissions & resourcetype parameters in listblob response  (Azure#23687)

* one

* Add permissions in ListBlobsIncludeItems

* Add permissions in ListBlobsIncludeItems

* Appending the value to swagger

* Added Owner,Group,Permissions,Acl,ResourceType in ListBlob Response

* [Storage] Prep for stg 95 GA release (Azure#23742)

* Prep for stg 95 GA release

* Update CHANGELOG.md

format for changelog

* Add SparseCheckoutPaths to archetype-typespec-emitter.yml (Azure#23740)

Co-authored-by: Wei Hu <[email protected]>

* [azopenai] Update azopenai changelog. (Azure#23732)

* Fix missing stress matrix display name filter (Azure#23726)

Co-authored-by: Ben Broderick Phillips <[email protected]>

* update date (Azure#23746)

* [azopenai] non-existing deployments fail in a different way now. (Azure#23747)

* Increment package version after release of ai/azopenai (Azure#23750)

* [Release] sdk/resourcemanager/hybridcompute/armhybridcompute/2.1.0-beta.1 generation from spec commit: 63d530d0def1c624f5d42d39170ff4ac196522e2 (Azure#23753)

* Increment package version after release of security/keyvault/azadmin (Azure#23749)

* Update azopenai* to use hardcoded variable group config (Azure#23748)

* Update azopenai* to use hardcoded variable group config

* Remove bicep template for ai services

* [Release] sdk/resourcemanager/networkcloud/armnetworkcloud/1.2.0-beta.1 generation from spec commit: 4e46346c63935bace7e24f806b4b084f2589d660 (Azure#23757)

* Add helpers to log a GitHub "notice" (Azure#23764)

Co-authored-by: Mike Harder <[email protected]>

* Increment package version after release of storage/azdatalake (Azure#23745)

* Increment package version after release of storage/azblob (Azure#23744)

* Increment package version after release of storage/azfile (Azure#23743)

* batching adjustments for create-prjobmatrix (Azure#23773)

Co-authored-by: Scott Beddall <[email protected]>

* changelog

* updated version

* Update sdk/ai/azopenaiassistants/CHANGELOG.md

Co-authored-by: Wes Haggard <[email protected]>

* updated

---------

Co-authored-by: ripark <[email protected]>
Co-authored-by: Richard Park <[email protected]>
Co-authored-by: Azure SDK Bot <[email protected]>
Co-authored-by: Daniel Jurek <[email protected]>
Co-authored-by: Ben Broderick Phillips <[email protected]>
Co-authored-by: Chenjie Shi <[email protected]>
Co-authored-by: Ashley Stanton-Nurse <[email protected]>
Co-authored-by: Scott Beddall <[email protected]>
Co-authored-by: Scott Beddall <[email protected]>
Co-authored-by: Patrick Hallisey <[email protected]>
Co-authored-by: ashruti-msft <[email protected]>
Co-authored-by: tanyasethi-msft <[email protected]>
Co-authored-by: Wei Hu <[email protected]>
Co-authored-by: Grace Wilcox <[email protected]>
Co-authored-by: Judy Liu <[email protected]>
Co-authored-by: Mike Harder <[email protected]>
Co-authored-by: Wes Haggard <[email protected]>

Author: 17 people

sdk/ai/azopenaiassistants/CHANGELOG.md

@@ -1,5 +1,15 @@
 # Release History
 
+## 0.3.0 (2024-12-03)
+
+### Features Added
+
+- Added support for the `ParallelToolCalls` option.
+
+### Breaking Changes
+
+- `MessageAttachmentToolAssignment` is now `MessageAttachmentToolDefinition`.
+
 ## 0.2.1 (2024-09-10)
 
 ### Features Added
@@ -23,7 +33,6 @@
 ### Bugs Fixed
 
 - ThreadRun.RequiredAction was deserialized incorrectly, making it impossible to actually resubmit a tool output. (PR#22834)
-
 ## 0.1.0 (2024-03-05)
 
 * Initial release of the `azopenaiassistants` library

sdk/ai/azopenaiassistants/autorest.md

@@ -155,6 +155,22 @@ directive:
       return $;
 ```
 
+MessageAttachmentToolDefinition
+
+```yaml
+directive:
+  - from: swagger-document
+    where: $.definitions
+    transform: |
+      // create the dummy type that I can use to point to my manually created union
+      $.MessageAttachmentToolDefinition = {
+        "x-ms-external": true,
+        "type": "object",
+        "properties": { "ignored": { "type": "string" } },
+        "x-ms-client-name": "MessageAttachmentToolDefinition"
+      };
+```
+
 ## Model -> DeploymentName
 
 ```yaml

sdk/ai/azopenaiassistants/client.go

@@ -157,9 +157,9 @@ func TestAssistantMessages(t *testing.T) {
 			Attachments: []azopenaiassistants.MessageAttachment{
 				{
 					FileID: uploadResp.ID,
-					Tools: []azopenaiassistants.MessageAttachmentToolAssignment{
-						{Type: to.Ptr(azopenaiassistants.MessageAttachmentToolAssignmentTypeCodeInterpreter)},
-						{Type: to.Ptr(azopenaiassistants.MessageAttachmentToolAssignmentTypeFileSearch)},
+					Tools: []azopenaiassistants.MessageAttachmentToolDefinition{
+						{CodeInterpreterToolDefinition: &azopenaiassistants.CodeInterpreterToolDefinition{}},
+						{FileSearchToolDefinition: &azopenaiassistants.FileSearchToolDefinition{}},
 					},
 				},
 			},
@@ -169,17 +169,12 @@ func TestAssistantMessages(t *testing.T) {
 		attachmentTools := messageResp.Attachments[0].Tools
 
 		// just trying to keep a consistent ordering of the tools for our checks.
-		if *attachmentTools[0].Type == azopenaiassistants.MessageAttachmentToolAssignmentTypeFileSearch {
+		if attachmentTools[0].FileSearchToolDefinition != nil {
 			attachmentTools[0], attachmentTools[1] = attachmentTools[1], attachmentTools[0]
 		}
 
-		require.Equal(t, azopenaiassistants.MessageAttachmentToolAssignment{
-			Type: to.Ptr(azopenaiassistants.MessageAttachmentToolAssignmentType("code_interpreter")),
-		}, attachmentTools[0])
-
-		require.Equal(t, azopenaiassistants.MessageAttachmentToolAssignment{
-			Type: to.Ptr(azopenaiassistants.MessageAttachmentToolAssignmentType("file_search")),
-		}, attachmentTools[1])
+		require.NotNil(t, attachmentTools[0].CodeInterpreterToolDefinition)
+		require.NotNil(t, attachmentTools[1].FileSearchToolDefinition)
 
 		messageID := messageResp.ID
 

sdk/ai/azopenaiassistants/client_test.go

@@ -266,24 +266,8 @@ func (t *transformer) replaceDocs() error {
 	log.Printf("replaceDocs()")
 
 	err := transformFiles(t.fileCache, "replaceDocs", []string{"client.go"}, func(text string) (string, error) {
-		docString := `
-// CreateVectorStoreFile - Create a vector store file by attaching a file to a vector store.
-// If the operation fails it returns an *azcore.ResponseError type.
-//
-// Generated from API version 2024-07-01-preview
-//   - vectorStoreID - The ID of the vector store for which to create a File.
-//   - options - CreateVectorStoreFileOptions contains the optional parameters for the Client.CreateVectorStoreFile method.
-`
-		newDocString := `
-// CreateVectorStoreFile - Create a vector store file by attaching a file to a vector store.
-// If the operation fails it returns an *azcore.ResponseError type.
-//
-// Generated from API version 2024-07-01-preview
-//   - vectorStoreID - The ID of the vector store for which to create a File.
-//   - body - Request object for creating a vector store file.
-//   - options - CreateVectorStoreFileOptions contains the optional parameters for the Client.CreateVectorStoreFile method.
-`
-		text = strings.ReplaceAll(text, docString, newDocString)
+		docStringRE := regexp.MustCompile(`(?s)(- options - CreateVectorStoreFileOptions)`)
+		text = docStringRE.ReplaceAllString(text, "  - body - Request object for creating a vector store file.\n"+`//  $1`)
 		return text, nil
 	}, nil)
 	return err

sdk/ai/azopenaiassistants/constants.go

@@ -5,6 +5,7 @@ package azopenaiassistants
 
 import (
 	"encoding/json"
+	"errors"
 	"fmt"
 )
 
@@ -151,3 +152,47 @@ func unmarshalStringOrObject[T any](jsonBytes []byte) (string, *T, error) {
 
 	return "", model, nil
 }
+
+// MessageAttachmentToolDefinition allows you to specify tools for use with a message attachment.
+type MessageAttachmentToolDefinition struct {
+	*CodeInterpreterToolDefinition
+	*FileSearchToolDefinition
+}
+
+// MarshalJSON implements the json.Marshaller interface for type MessageAttachmentToolDefinition.
+func (m MessageAttachmentToolDefinition) MarshalJSON() ([]byte, error) {
+	if m.CodeInterpreterToolDefinition != nil && m.FileSearchToolDefinition != nil {
+		return nil, errors.New("only one tool definition should be set in MessageAttachmentToolDefinition")
+	}
+
+	switch {
+	case m.CodeInterpreterToolDefinition != nil:
+		return json.Marshal(m.CodeInterpreterToolDefinition)
+	case m.FileSearchToolDefinition != nil:
+		return json.Marshal(m.FileSearchToolDefinition)
+	default:
+		return nil, errors.New("no tool definition was set in MessageAttachmentToolDefinition")
+	}
+}
+
+// UnmarshalJSON implements the json.Marshaller interface for type MessageAttachmentToolDefinition.
+func (m *MessageAttachmentToolDefinition) UnmarshalJSON(data []byte) error {
+	// There's only two types right now (CodeInterpreterToolDefinition and FileSearchToolDefinition)
+	// and CodeInterpreterToolDefinition is a subset of FileSearchToolDefinition
+	var toolDef *FileSearchToolDefinition
+
+	if err := json.Unmarshal(data, &toolDef); err != nil {
+		return err
+	}
+
+	switch *toolDef.Type {
+	case "code_interpreter":
+		m.CodeInterpreterToolDefinition = &CodeInterpreterToolDefinition{Type: toolDef.Type}
+		return nil
+	case "file_search":
+		m.FileSearchToolDefinition = toolDef
+		return nil
+	default:
+		return fmt.Errorf("unhandled tool definition type %s", *toolDef.Type)
+	}
+}

sdk/ai/azopenaiassistants/internal/transform/transform.go

@@ -1,5 +1,5 @@
 #location: https://raw.githubusercontent.com/Azure/azure-rest-api-specs/cacb3dc095486d8691c47dea944fc9ed0f4d0e32/specification/ai/OpenAI.Assistants/client.tsp
 directory: specification/ai/OpenAI.Assistants
 # https://github.com/Azure/azure-rest-api-specs/pull/29067
-commit: eb46d197d57e1f71f1819f37779f387f20c344e9
+commit: ea5bac870d87ff0424a4e4f5c448b0a66890838d
 repo: Azure/azure-rest-api-specs