Method Connect

examples/docker_agent/main.go:42–163

(stream grpc.BidiStreamingServer[proto.AgentMessage, proto.AgentMessage])

}

func (s *server) Connect(stream grpc.BidiStreamingServer[proto.AgentMessage, proto.AgentMessage]) error {
	incoming, err := stream.Recv()
	if err == io.EOF {
		return nil
	}
	if err != nil {
		return err
	}

	start := incoming.GetStart()
	if start == nil {
		return errors.New("missing start message")
	}

	contents := protoToContents(start.Messages)

	for {
		model := os.Getenv("AX_GEMINI_MODEL")
		if model == "" {
			model = "gemini-3.5-flash"
		}
		resp, err := s.genaiClient.Models.GenerateContent(stream.Context(), model, contents, &genai.GenerateContentConfig{
			SystemInstruction: genai.Text("You are DockerAgent, an agent specialized in writing Dockerfiles. Use the docker-writer skill to fulfill requests. Keep responses concise.")[0],
			Tools:             []*genai.Tool{skills.BuildTool(s.skillsExecutor.SkillNames())},
		})
		if err != nil {
			return fmt.Errorf("failed to generate content: %w", err)
		}

		if len(resp.Candidates) == 0 {
			return errors.New("no candidates from Gemini")
		}
		candidate := resp.Candidates[0]

		var keepLooping bool
		var functionCall *genai.FunctionCall
		var thoughtSignature []byte

		for _, part := range candidate.Content.Parts {
			if part.FunctionCall != nil {
				functionCall = part.FunctionCall
				thoughtSignature = part.ThoughtSignature
				break
			}
		}

		if functionCall != nil {
			if functionCall.ID == "" {
				functionCall.ID = uuid.NewString()
			}

			// Handle skill call
			result, err := s.skillsExecutor.HandleCall(stream.Context(), functionCall)
			if err != nil {
				return fmt.Errorf("failed to handle skill call: %w", err)
			}

			// Append function call and response to history