cyberia/research/mimi/internal/bot/llm/agent/logseqquery/logseqquery.go

go
ฯ€ 0.0% 181 words ยท 1 min 
package logseqquery

import (
	"bytes"
	"context"
	"encoding/csv"
	"fmt"
	"log/slog"

	"github.com/firebase/genkit/go/ai"

	"mimi/internal/bot/llm/agent"
	"mimi/internal/provider/logseq"
	"mimi/internal/provider/logseq/query"
)

type LogseqQueryAgent struct {
	graph logseq.RegexGraph
}

func New(graph logseq.RegexGraph) LogseqQueryAgent {
	return LogseqQueryAgent{graph: graph}
}

func (a LogseqQueryAgent) GetInfo() agent.Info {
	return agent.Info{
		Name: "logseq-query",
		Description: `Evaluates logseq queries of format like '<div class="query-results"><div class="query-results-header">Query: <code>(property :supply &quot;next-month&quot;)</code></div><p><em>No results</em></p></div>'. They can have additional parameters like '<div class="query-results"><div class="query-results-header">Query: <code>(page-tags super)</code></div><p><em>No results</em></p></div>
  query-properties:: [:page :tags :alias]
  query-sort-by:: page
	query-sort-desc:: true'

		As a result it returns a CSV file with results`,
	}
}

func (a LogseqQueryAgent) Run(ctx context.Context, queryS string, msgs ...*ai.Message) (agent.Response, error) {
	var result agent.Response
	slog.Info("trying to eval logseq query")
	out, err := query.Eval(ctx, a.graph, queryS)
	if err != nil {
		slog.Warn("failed to evaluate logseq query", "with", err)
		return result, fmt.Errorf("failed to evaluate query with %w", err)
	}
	slog.Info("got logseq query result", "rows", len(out.Table)-1)

	// Encode as CSV
	buf := new(bytes.Buffer)
	writer := csv.NewWriter(buf)
	err = writer.WriteAll(out.Table)
	if err != nil {
		return result, fmt.Errorf("failed to encode LogSeq query results into CSV with %w", err)
	}

	result = agent.NewResponse(agent.DataFile{
		Blob: buf.Bytes(),
		Name: "query-result.csv",
	}, nil)
	return result, nil
}

Graph