Upgraded SDK

Author: nfx

go-libs/llnotes/gen.go

@@ -3,11 +3,14 @@ package llnotes
 import (
 	"context"
 	"fmt"
+	"strings"
 
+	"github.com/databricks/databricks-sdk-go"
 	"github.com/databricks/databricks-sdk-go/client"
 	"github.com/databricks/databricks-sdk-go/config"
 	"github.com/databricks/databricks-sdk-go/listing"
 	"github.com/databricks/databricks-sdk-go/logger"
+	"github.com/databricks/databricks-sdk-go/service/serving"
 	"github.com/databrickslabs/sandbox/go-libs/github"
 )
 
@@ -18,7 +21,7 @@ type ChangeDetector struct {
 	Model      string
 }
 
-func (c *ChangeDetector) Run(ctx context.Context) ([]any, error) {
+func (c *ChangeDetector) Run(ctx context.Context) ([]string, error) {
 	versions, err := listing.ToSlice(ctx, c.GitHub.Versions(ctx, c.Org, c.Repo))
 	if err != nil {
 		return nil, fmt.Errorf("versions: %w", err)
@@ -39,11 +42,39 @@ func (c *ChangeDetector) Run(ctx context.Context) ([]any, error) {
 	if c.Model == "" {
 		c.Model = "databricks-mixtral-8x7b-instruct"
 	}
-	client, err := client.New(c.Databricks)
+	return c.tryChat2(ctx, commits)
+}
+
+func (c *ChangeDetector) tryChat2(ctx context.Context, commits []github.RepositoryCommit) ([]string, error) {
+	w, err := databricks.NewWorkspaceClient((*databricks.Config)(c.Databricks))
 	if err != nil {
-		return nil, fmt.Errorf("databricks client: %w", err)
+		return nil, err
+	}
+	x := []string{}
+	for _, c := range commits {
+		x = append(x, c.Commit.Message)
+	}
+	response, err := w.ServingEndpoints.Query(ctx, serving.QueryEndpointInput{
+		Name: c.Model,
+		Messages: []serving.ChatMessage{
+			{
+				Role:    "system",
+				Content: blogPrompt,
+			},
+			{
+				Role:    "user",
+				Content: strings.Join(x, "\n"),
+			},
+		},
+	})
+	if err != nil {
+		return nil, err
+	}
+	out := []string{}
+	for _, v := range response.Choices {
+		out = append(out, v.Message.Content)
 	}
-	return []any{}, c.tryChat(ctx, client, commits)
+	return out, nil
 }
 
 func (c *ChangeDetector) tryChat(ctx context.Context, client *client.DatabricksClient, commits []github.RepositoryCommit) error {

go-libs/llnotes/prompts.go

@@ -0,0 +1,18 @@
+package llnotes
+
+const prPrompt = `You are a staff software engineer, and you are reviewing code in a pull request. If new methods are added, explain what these methods are doing. 
+If this pull request has changes to tests, give a summary of scenarios that are tested. Please summarize this pull request in one paragraph. 
+Changes related to tests should be in a separate paragraph.`
+
+const blogPrompt = `SYSTEM:
+You are professional technical writer and you receive draft release notes for v0.11.0 of project called UCX by Databricks Labs in a markdown format from multiple team members. ucx project can be described as "Your best companion for upgrading to Unity Catalog. UCX will guide you, the Databricks customer, through the process of upgrading your account, groups, workspaces, jobs etc. to Unity Catalog."
+
+You write a long post announcement that takes 5 minutes to read, summarize the most important features, and mention them on top. Keep the markdown links when relevant.
+Do not use headings. Write fluent paragraphs, that are at least few sentences long. Blog post title should nicely summarize the feature increments of this release.
+
+Don't abuse lists. paragraphs should have at least 3-4 sentences. The title should be one-sentence summary of the incremental updates for this release
+
+You aim at driving more adoption of the project on Medium.
+
+USER:
+`