add off topic, funny, and check quotes

handle_post.go

···
 
 import (
 	"context"
-	"errors"
 	"fmt"
 	"time"
 
···
 		return nil
 	}
 
-	if post.Reply == nil {
-		return nil
+	var parentUri string
+
+	if post.Reply != nil && post.Reply.Parent != nil {
+		parentUri = post.Reply.Parent.Uri
+	} else if post.Embed != nil && post.Embed.EmbedRecord != nil && post.Embed.EmbedRecord.Record != nil {
+		parentUri = post.Embed.EmbedRecord.Record.Uri
+	} else if post.Embed != nil && post.Embed.EmbedRecordWithMedia != nil && post.Embed.EmbedRecordWithMedia.Record != nil && post.Embed.EmbedRecordWithMedia.Record.Record != nil {
+		parentUri = post.Embed.EmbedRecordWithMedia.Record.Record.Uri
 	}
 
-	if post.Reply.Parent == nil {
-		return errors.New("badly formatted reply ref (no parent)")
+	if parentUri == "" {
+		return nil
 	}
 
-	atUri, err := syntax.ParseATURI(post.Reply.Parent.Uri)
+	atUri, err := syntax.ParseATURI(parentUri)
 	if err != nil {
 		return fmt.Errorf("failed to parse parent aturi: %w", err)
 	}
···
 		return nil
 	}
 
-	parent, err := dsmt.getPost(ctx, post.Reply.Parent.Uri)
+	parent, err := dsmt.getPost(ctx, parentUri)
 	if err != nil {
 		return fmt.Errorf("failed to get parent post: %w", err)
 	}
···
 	ctx, cancel := context.WithTimeout(ctx, 30*time.Second)
 	defer cancel()
 
-	isBadFaith, err := dsmt.lmstudioc.GetIsBadFaith(ctx, parent.Text, post.Text)
+	results, err := dsmt.lmstudioc.GetIsBadFaith(ctx, parent.Text, post.Text)
 	if err != nil {
 		return fmt.Errorf("failed to check bad faith: %w", err)
 	}
 
-	if !isBadFaith {
-		logger.Info("determined that reply was not bad faith")
-		return nil
+	if results.BadFaith {
+		if err := dsmt.emitLabel(ctx, uri, LabelBadFaith); err != nil {
+			return fmt.Errorf("failed to label post: %w", err)
+		}
+		logger.Info("determined that reply was bad faith and emitted label")
 	}
 
-	if err := dsmt.emitLabel(ctx, uri, LabelBadFaith); err != nil {
-		return fmt.Errorf("failed to label post: %w", err)
+	if results.OffTopic {
+		if err := dsmt.emitLabel(ctx, uri, LabelOffTopic); err != nil {
+			return fmt.Errorf("failed to label post: %w", err)
+		}
+		logger.Info("determined that reply was off topic and emitted label")
 	}
 
-	logger.Info("determined that reply was bad faith and emitted label")
+	if results.Funny {
+		if err := dsmt.emitLabel(ctx, uri, LabelFunny); err != nil {
+			return fmt.Errorf("failed to label post: %w", err)
+		}
+		logger.Info("determined that reply was funny and emitted label")
+	}
 
 	return nil
 }

labeler/index.ts

···
 
 const LABELS: Record<string, boolean> = {
   'bad-faith': true,
+  'off-topic': true,
+  funny: true,
 }
 
 function run() {

lmstudio.go

···
 				Type:        "boolean",
 				Description: "Whether the reply to the parent is bad faith or not.",
 			},
+			"off_topic": {
+				Type:        "boolean",
+				Description: "Whether the reply to the parent is off topic.",
+			},
+			"funny": {
+				Type:        "boolean",
+				Description: "Whether the reply to the parent is funny.",
+			},
 		},
-		Required: []string{"bad_faith"},
+		Required: []string{"bad_faith", "off_topic", "funny"},
 	}
 )
 
···
 	return &chatResp, nil
 }
 
-func (c *LMStudioClient) GetIsBadFaith(ctx context.Context, parent, reply string) (bool, error) {
+type BadFaithResults struct {
+	BadFaith bool
+	OffTopic bool
+	Funny    bool
+}
+
+func (c *LMStudioClient) GetIsBadFaith(ctx context.Context, parent, reply string) (*BadFaithResults, error) {
 	request := ChatRequest{
 		Model: "google/gemma-3-27b",
 		Messages: []Message{
 			{
 				Role:    "system",
-				Content: "You are an observer of posts on a microblogging website. You determine if the second message provided by the user is a bad faith reply to the second message provided to you. Opposing viewpoints are good, and should be appreciated. However, things that are toxic, trollish, or offer no good value to the conversation are considered bad faith.",
+				Content: "You are an observer of posts on a microblogging website. You determine if the second message provided by the user is a bad faith reply, and off topic reply, and/or a funny reply to the second message provided to you. Opposing viewpoints are good, and should be appreciated. However, things that are toxic, trollish, or offer no good value to the conversation are considered bad faith. Just because something is bad faith or off topic does not mean the post cannot also be funny.",
 			},
 			{
 				Role:    "user",
···
 			},
 		},
 		Temperature: 0.7,
-		MaxTokens:   50,
+		MaxTokens:   100,
 		ResponseFormat: &ResponseFormat{
 			Type: "json_schema",
 			JSONSchema: &JSONSchemaWrap{
···
 	}
 	response, err := c.sendChatRequest(request)
 	if err != nil {
-		return false, fmt.Errorf("failed to get chat response: %w", err)
+		return nil, fmt.Errorf("failed to get chat response: %w", err)
 	}
 
 	var result map[string]any
 	if err := json.Unmarshal([]byte(response.Choices[0].Message.Content), &result); err != nil {
-		return false, fmt.Errorf("failed to unmarshal response: %w", err)
+		return nil, fmt.Errorf("failed to unmarshal response: %w", err)
 	}
 
 	badFaith, ok := result["bad_faith"].(bool)
 	if !ok {
-		return false, fmt.Errorf("model gave bad response, not structured")
+		return nil, fmt.Errorf("model gave bad response (bad faith), not structured")
 	}
 
-	return badFaith, nil
+	offTopic, ok := result["off_topic"].(bool)
+	if !ok {
+		return nil, fmt.Errorf("model gave bad response (off topic), not structured")
+	}
+
+	funny, ok := result["funny"].(bool)
+	if !ok {
+		return nil, fmt.Errorf("model gave bad response (funny), not structured")
+	}
+
+	return &BadFaithResults{
+		BadFaith: badFaith,
+		OffTopic: offTopic,
+		Funny:    funny,
+	}, nil
 }

main.go

···
 
 const (
 	LabelBadFaith = "bad-faith"
+	LabelOffTopic = "off-topic"
+	LabelFunny    = "funny"
 )
 
 func main() {