[AI Agents Extension] Tool parsing (#6091)

* Add tool resource parsing for hosted

Signed-off-by: trangevi <[email protected]>

* move the handling

Signed-off-by: trangevi <[email protected]>

* Add support for prompt tools handling

Signed-off-by: trangevi <[email protected]>

* cspell

Signed-off-by: trangevi <[email protected]>

* capitalization

Signed-off-by: trangevi <[email protected]>

* cspell again

Signed-off-by: trangevi <[email protected]>

* Add check for connection id from registry

Signed-off-by: trangevi <[email protected]>

---------

Signed-off-by: trangevi <[email protected]>

Author: trangevi

cli/azd/.vscode/cspell.yaml

@@ -188,6 +188,14 @@ overrides:
       - mcpservertoolspecifyapprovalmode
       - protocolversionrecord
       - openapitool
+  - filename: extensions/azure.ai.agents/internal/pkg/agents/agent_api/models.go
+    words:
+      - Dataagent
+      - dataagent
+      - webp
+  - filename: extensions/azure.ai.agents/internal/pkg/agents/registry_api/operations.go
+    words:
+      - Dataagent
   - filename: docs/new-azd-command.md
     words:
       - pflag

cli/azd/extensions/azure.ai.agents/internal/cmd/init.go

@@ -721,6 +721,14 @@ func (a *InitAction) downloadAgentYaml(
 		return nil, "", fmt.Errorf("failed to process manifest parameters: %w", err)
 	}
 
+	_, isPromptAgent := agentManifest.Template.(agent_yaml.PromptAgent)
+	if isPromptAgent {
+		agentManifest, err = agent_yaml.ProcessPromptAgentToolsConnections(ctx, agentManifest, a.azdClient)
+		if err != nil {
+			return nil, "", fmt.Errorf("failed to process prompt agent tools connections: %w", err)
+		}
+	}
+
 	content, err = yaml.Marshal(agentManifest)
 	if err != nil {
 		return nil, "", fmt.Errorf("marshaling agent manifest to YAML after parameter processing: %w", err)
@@ -800,6 +808,7 @@ func (a *InitAction) addToProject(ctx context.Context, targetDir string, agentMa
 	var agentConfig = project.ServiceTargetAgentConfig{}
 
 	deploymentDetails := []project.Deployment{}
+	resourceDetails := []project.Resource{}
 	switch agentDef.Kind {
 	case agent_yaml.AgentKindPrompt:
 		agentDef := agentManifest.Template.(agent_yaml.PromptAgent)
@@ -836,9 +845,39 @@ func (a *InitAction) addToProject(ctx context.Context, targetDir string, agentMa
 				deploymentDetails = append(deploymentDetails, *modelDeployment)
 			}
 		}
+
+		// Handle tool resources that require connection names
+		if agentManifest.Resources != nil {
+			for _, resource := range agentManifest.Resources {
+				// Try to cast to ToolResource
+				if toolResource, ok := resource.(agent_yaml.ToolResource); ok {
+					// Check if this is a resource that requires a connection name
+					if toolResource.Id == "bing_grounding" || toolResource.Id == "azure_ai_search" {
+						// Prompt the user for a connection name
+						resp, err := a.azdClient.Prompt().Prompt(ctx, &azdext.PromptRequest{
+							Options: &azdext.PromptOptions{
+								Message:        fmt.Sprintf("Enter connection name for %s resource:", toolResource.Id),
+								IgnoreHintKeys: true,
+							},
+						})
+						if err != nil {
+							return fmt.Errorf("prompting for connection name for %s: %w", toolResource.Id, err)
+						}
+
+						// Add to resource details
+						resourceDetails = append(resourceDetails, project.Resource{
+							Resource:       toolResource.Id,
+							ConnectionName: resp.Value,
+						})
+					}
+				}
+				// Skip the resource if the cast fails
+			}
+		}
 	}
 
 	agentConfig.Deployments = deploymentDetails
+	agentConfig.Resources = resourceDetails
 
 	var agentConfigStruct *structpb.Struct
 	if agentConfigStruct, err = project.MarshalStruct(&agentConfig); err != nil {

cli/azd/extensions/azure.ai.agents/internal/cmd/listen.go

@@ -96,6 +96,10 @@ func preprovisionEnvUpdate(ctx context.Context, azdClient *azdext.AzdClient, azd
 		return err
 	}
 
+	if err := resourcesEnvUpdate(ctx, foundryAgentConfig.Resources, azdClient, currentEnvResponse.Environment.Name); err != nil {
+		return err
+	}
+
 	return nil
 }
 
@@ -163,6 +167,20 @@ func deploymentEnvUpdate(ctx context.Context, deployments []project.Deployment,
 	return setEnvVar(ctx, azdClient, envName, "AI_PROJECT_DEPLOYMENTS", escapedJsonString)
 }
 
+func resourcesEnvUpdate(ctx context.Context, resources []project.Resource, azdClient *azdext.AzdClient, envName string) error {
+	resourcesJson, err := json.Marshal(resources)
+	if err != nil {
+		return fmt.Errorf("failed to marshal resource details to JSON: %w", err)
+	}
+
+	// Escape backslashes and double quotes for environment variable
+	jsonString := string(resourcesJson)
+	escapedJsonString := strings.ReplaceAll(jsonString, "\\", "\\\\")
+	escapedJsonString = strings.ReplaceAll(escapedJsonString, "\"", "\\\"")
+
+	return setEnvVar(ctx, azdClient, envName, "AI_PROJECT_DEPENDENT_RESOURCES", escapedJsonString)
+}
+
 func containerAgentHandling(ctx context.Context, azdClient *azdext.AzdClient, project *azdext.ProjectConfig, svc *azdext.ServiceConfig) error {
 	servicePath := svc.RelativePath
 	fullPath := filepath.Join(project.Path, servicePath)