@dcode-dev/dcode-cli 1.0.0

package/go.sum

@@ -0,0 +1,86 @@
+al.essio.dev/pkg/shellescape v1.5.1 h1:86HrALUujYS/h+GtqoB26SBEdkWfmMI6FubjXlsXyho=
+al.essio.dev/pkg/shellescape v1.5.1/go.mod h1:6sIqp7X2P6mThCQ7twERpZTuigpr6KbZWtls1U8I890=
+cloud.google.com/go/compute/metadata v0.9.0 h1:pDUj4QMoPejqq20dK0Pg2N4yG9zIkYGdBtwLoEkH9Zs=
+cloud.google.com/go/compute/metadata v0.9.0/go.mod h1:E0bWwX5wTnLPedCKqk3pJmVgCBSM6qQI1yTBdEb3C10=
+github.com/atotto/clipboard v0.1.4 h1:EH0zSVneZPSuFR11BlR9YppQTVDbh5+16AmcJi4g1z4=
+github.com/atotto/clipboard v0.1.4/go.mod h1:ZY9tmq7sm5xIbd9bOK4onWV4S6X0u6GY7Vn0Yu86PYI=
+github.com/aymanbagabas/go-osc52/v2 v2.0.1 h1:HwpRHbFMcZLEVr42D4p7XBqjyuxQH5SMiErDT4WkJ2k=
+github.com/aymanbagabas/go-osc52/v2 v2.0.1/go.mod h1:uYgXzlJ7ZpABp8OJ+exZzJJhRNQ2ASbcXHWsFqH8hp8=
+github.com/charmbracelet/bubbles v0.18.0 h1:PYv1A036luoBGroX6VWjQIE9Syf2Wby2oOl/39KLfy0=
+github.com/charmbracelet/bubbles v0.18.0/go.mod h1:08qhZhtIwzgrtBjAcJnij1t1H0ZRjwHyGsy6AL11PSw=
+github.com/charmbracelet/bubbletea v0.25.0 h1:bAfwk7jRz7FKFl9RzlIULPkStffg5k6pNt5dywy4TcM=
+github.com/charmbracelet/bubbletea v0.25.0/go.mod h1:EN3QDR1T5ZdWmdfDzYcqOCAps45+QIJbLOBxmVNWNNg=
+github.com/charmbracelet/lipgloss v0.9.1 h1:PNyd3jvaJbg4jRHKWXnCj1akQm4rh8dbEzN1p/u1KWg=
+github.com/charmbracelet/lipgloss v0.9.1/go.mod h1:1mPmG4cxScwUQALAAnacHaigiiHB9Pmr+v1VEawJl6I=
+github.com/containerd/console v1.0.4-0.20230313162750-1ae8d489ac81 h1:q2hJAaP1k2wIvVRd/hEHD7lacgqrCPS+k8g1MndzfWY=
+github.com/containerd/console v1.0.4-0.20230313162750-1ae8d489ac81/go.mod h1:YynlIjWYF8myEu6sdkwKIvGQq+cOckRm6So2avqoYAk=
+github.com/cpuguy83/go-md2man/v2 v2.0.3/go.mod h1:tgQtvFlXSQOSOSIRvRPT7W67SCa46tRHOmNcaadrF8o=
+github.com/danieljoos/wincred v1.2.2 h1:774zMFJrqaeYCK2W57BgAem/MLi6mtSE47MB6BOJ0i0=
+github.com/danieljoos/wincred v1.2.2/go.mod h1:w7w4Utbrz8lqeMbDAK0lkNJUv5sAOkFi7nd/ogr0Uh8=
+github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
+github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
+github.com/godbus/dbus/v5 v5.1.0 h1:4KLkAxT3aOY8Li4FRJe/KvhoNFFxo0m6fNuFUO8QJUk=
+github.com/godbus/dbus/v5 v5.1.0/go.mod h1:xhWf0FNVPg57R7Z0UbKHbJfkEywrmjJnf7w5xrFpKfA=
+github.com/google/go-cmp v0.7.0 h1:wk8382ETsv4JYUZwIsn6YpYiWiBsYLSJiTsyBybVuN8=
+github.com/google/go-cmp v0.7.0/go.mod h1:pXiqmnSA92OHEEa9HXL2W4E7lf9JzCmGVUdgjX3N/iU=
+github.com/google/shlex v0.0.0-20191202100458-e7afc7fbc510 h1:El6M4kTTCOh6aBiKaUGG7oYTSPP8MxqL4YI3kZKwcP4=
+github.com/google/shlex v0.0.0-20191202100458-e7afc7fbc510/go.mod h1:pupxD2MaaD3pAXIBCelhxNneeOaAeabZDe5s4K6zSpQ=
+github.com/inconshreveable/mousetrap v1.1.0 h1:wN+x4NVGpMsO7ErUn/mUI3vEoE6Jt13X2s0bqwp9tc8=
+github.com/inconshreveable/mousetrap v1.1.0/go.mod h1:vpF70FUmC8bwa3OWnCshd2FqLfsEA9PFc4w1p2J65bw=
+github.com/joho/godotenv v1.5.1 h1:7eLL/+HRGLY0ldzfGMeQkb7vMd0as4CfYvUVzLqw0N0=
+github.com/joho/godotenv v1.5.1/go.mod h1:f4LDr5Voq0i2e/R5DDNOoa2zzDfwtkZa6DnEwAbqwq4=
+github.com/lucasb-eyer/go-colorful v1.2.0 h1:1nnpGOrhyZZuNyfu1QjKiUICQ74+3FNCN69Aj6K7nkY=
+github.com/lucasb-eyer/go-colorful v1.2.0/go.mod h1:R4dSotOR9KMtayYi1e77YzuveK+i7ruzyGqttikkLy0=
+github.com/mattn/go-isatty v0.0.20 h1:xfD0iDuEKnDkl03q4limB+vH+GxLEtL/jb4xVJSWWEY=
+github.com/mattn/go-isatty v0.0.20/go.mod h1:W+V8PltTTMOvKvAeJH7IuucS94S2C6jfK/D7dTCTo3Y=
+github.com/mattn/go-localereader v0.0.1 h1:ygSAOl7ZXTx4RdPYinUpg6W99U8jWvWi9Ye2JC/oIi4=
+github.com/mattn/go-localereader v0.0.1/go.mod h1:8fBrzywKY7BI3czFoHkuzRoWE9C+EiG4R1k4Cjx5p88=
+github.com/mattn/go-runewidth v0.0.3/go.mod h1:LwmH8dsx7+W8Uxz3IHJYH5QSwggIsqBzpuz5H//U1FU=
+github.com/mattn/go-runewidth v0.0.12/go.mod h1:RAqKPSqVFrSLVXbA8x7dzmKdmGzieGRCM46jaSJTDAk=
+github.com/mattn/go-runewidth v0.0.15 h1:UNAjwbU9l54TA3KzvqLGxwWjHmMgBUVhBiTjelZgg3U=
+github.com/mattn/go-runewidth v0.0.15/go.mod h1:Jdepj2loyihRzMpdS35Xk/zdY8IAYHsh153qUoGf23w=
+github.com/muesli/ansi v0.0.0-20230316100256-276c6243b2f6 h1:ZK8zHtRHOkbHy6Mmr5D264iyp3TiX5OmNcI5cIARiQI=
+github.com/muesli/ansi v0.0.0-20230316100256-276c6243b2f6/go.mod h1:CJlz5H+gyd6CUWT45Oy4q24RdLyn7Md9Vj2/ldJBSIo=
+github.com/muesli/cancelreader v0.2.2 h1:3I4Kt4BQjOR54NavqnDogx/MIoWBFa0StPA8ELUXHmA=
+github.com/muesli/cancelreader v0.2.2/go.mod h1:3XuTXfFS2VjM+HTLZY9Ak0l6eUKfijIfMUZ4EgX0QYo=
+github.com/muesli/reflow v0.3.0 h1:IFsN6K9NfGtjeggFP+68I4chLZV2yIKsXJFNZ+eWh6s=
+github.com/muesli/reflow v0.3.0/go.mod h1:pbwTDkVPibjO2kyvBQRBxTWEEGDGq0FlB1BIKtnHY/8=
+github.com/muesli/termenv v0.15.2 h1:GohcuySI0QmI3wN8Ok9PtKGkgkFIk7y6Vpb5PvrY+Wo=
+github.com/muesli/termenv v0.15.2/go.mod h1:Epx+iuz8sNs7mNKhxzH4fWXGNpZwUaJKRS1noLXviQ8=
+github.com/pelletier/go-toml/v2 v2.2.4 h1:mye9XuhQ6gvn5h28+VilKrrPoQVanw5PMw/TB0t5Ec4=
+github.com/pelletier/go-toml/v2 v2.2.4/go.mod h1:2gIqNv+qfxSVS7cM2xJQKtLSTLUE9V8t9Stt+h56mCY=
+github.com/peterh/liner v1.2.2 h1:aJ4AOodmL+JxOZZEL2u9iJf8omNRpqHc/EbrK+3mAXw=
+github.com/peterh/liner v1.2.2/go.mod h1:xFwJyiKIXJZUKItq5dGHZSTBRAuG/CpeNpWLyiNRNwI=
+github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
+github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
+github.com/rivo/uniseg v0.1.0/go.mod h1:J6wj4VEh+S6ZtnVlnTBMWIodfgj8LQOQFoIToxlJtxc=
+github.com/rivo/uniseg v0.2.0/go.mod h1:J6wj4VEh+S6ZtnVlnTBMWIodfgj8LQOQFoIToxlJtxc=
+github.com/rivo/uniseg v0.4.7 h1:WUdvkW8uEhrYfLC4ZzdpI2ztxP1I582+49Oc5Mq64VQ=
+github.com/rivo/uniseg v0.4.7/go.mod h1:FN3SvrM+Zdj16jyLfmOkMNblXMcoc8DfTHruCPUcx88=
+github.com/russross/blackfriday/v2 v2.1.0/go.mod h1:+Rmxgy9KzJVeS9/2gXHxylqXiyQDYRxCVz55jmeOWTM=
+github.com/spf13/cobra v1.8.0 h1:7aJaZx1B85qltLMc546zn58BxxfZdR/W22ej9CFoEf0=
+github.com/spf13/cobra v1.8.0/go.mod h1:WXLWApfZ71AjXPya3WOlMsY9yMs7YeiHhFVlvLyhcho=
+github.com/spf13/pflag v1.0.5 h1:iy+VFUOCP1a+8yFto/drg2CJ5u0yRoB7fZw3DKv/JXA=
+github.com/spf13/pflag v1.0.5/go.mod h1:McXfInJRrz4CZXVZOBLb0bTZqETkiAhM9Iw0y3An2Bg=
+github.com/stretchr/objx v0.5.2 h1:xuMeJ0Sdp5ZMRXx/aWO6RZxdr3beISkG5/G/aIRr3pY=
+github.com/stretchr/objx v0.5.2/go.mod h1:FRsXN1f5AsAjCGJKqEizvkpNtU+EGNCLh3NxZ/8L+MA=
+github.com/stretchr/testify v1.11.1 h1:7s2iGBzp5EwR7/aIZr8ao5+dra3wiQyKjjFuvgVKu7U=
+github.com/stretchr/testify v1.11.1/go.mod h1:wZwfW3scLgRK+23gO65QZefKpKQRnfz6sD981Nm4B6U=
+github.com/zalando/go-keyring v0.2.6 h1:r7Yc3+H+Ux0+M72zacZoItR3UDxeWfKTcabvkI8ua9s=
+github.com/zalando/go-keyring v0.2.6/go.mod h1:2TCrxYrbUNYfNS/Kgy/LSrkSQzZ5UPVH85RwfczwvcI=
+golang.org/x/oauth2 v0.35.0 h1:Mv2mzuHuZuY2+bkyWXIHMfhNdJAdwW3FuWeCPYN5GVQ=
+golang.org/x/oauth2 v0.35.0/go.mod h1:lzm5WQJQwKZ3nwavOZ3IS5Aulzxi68dUSgRHujetwEA=
+golang.org/x/sync v0.19.0 h1:vV+1eWNmZ5geRlYjzm2adRgW2/mcpevXNg50YZtPCE4=
+golang.org/x/sync v0.19.0/go.mod h1:9KTHXmSnoGruLpwFjVSX0lNNA75CykiMECbovNTZqGI=
+golang.org/x/sys v0.0.0-20211117180635-dee7805ff2e1/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
+golang.org/x/sys v0.1.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
+golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
+golang.org/x/sys v0.40.0 h1:DBZZqJ2Rkml6QMQsZywtnjnnGvHza6BTfYFWY9kjEWQ=
+golang.org/x/sys v0.40.0/go.mod h1:OgkHotnGiDImocRcuBABYBEXf8A9a87e/uXjp9XT3ks=
+golang.org/x/term v0.39.0 h1:RclSuaJf32jOqZz74CkPA9qFuVTX7vhLlpfj/IGWlqY=
+golang.org/x/term v0.39.0/go.mod h1:yxzUCTP/U+FzoxfdKmLaA0RV1WgE0VY7hXBwKtY/4ww=
+golang.org/x/text v0.33.0 h1:B3njUFyqtHDUI5jMn1YIr5B0IE2U0qck04r6d4KPAxE=
+golang.org/x/text v0.33.0/go.mod h1:LuMebE6+rBincTi9+xWTY8TztLzKHc/9C1uBCG27+q8=
+gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
+gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
+gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=

package/internal/agent/agent.go

@@ -0,0 +1,332 @@
+package agent
+
+import (
+	"context"
+	"encoding/base64"
+	"fmt"
+	"os"
+	"path/filepath"
+	"strings"
+
+	"github.com/ddhanush1/dcode/internal/ai"
+	"github.com/ddhanush1/dcode/internal/websearch"
+)
+
+type Attachment struct {
+	Type       string `json:"type"`              // "image" supported
+	Mime       string `json:"mime,omitempty"`
+	DataBase64 string `json:"data_base64,omitempty"`
+}
+
+type WebSearchRequest struct {
+	Enabled bool   `json:"enabled"`
+	Query   string `json:"query,omitempty"`
+}
+
+type ChatRequest struct {
+	Messages   []ai.Message      `json:"messages"`
+	Attach     []Attachment      `json:"attachments,omitempty"`
+	WebSearch  WebSearchRequest  `json:"web_search,omitempty"`
+	ProjectDir string            `json:"project_dir,omitempty"` // optional, for future
+	Metadata   map[string]string `json:"metadata,omitempty"`
+}
+
+type WebResult struct {
+	Title string `json:"title"`
+	URL   string `json:"url"`
+	Snippet string `json:"snippet"`
+}
+
+type ChatResponse struct {
+	Assistant  string      `json:"assistant"`
+	WebResults []WebResult `json:"web_results,omitempty"`
+}
+
+type FileInput struct {
+	Path    string `json:"path"`
+	Content string `json:"content"`
+}
+
+type EditRequest struct {
+	Prompt      string      `json:"prompt"`
+	Files       []FileInput `json:"files"`
+	WebSearch   WebSearchRequest `json:"web_search,omitempty"`
+	Instructions string     `json:"instructions,omitempty"` // optional extra system guidance
+}
+
+type FileEdit struct {
+	Path       string `json:"path"`
+	NewContent string `json:"new_content"`
+}
+
+type EditResponse struct {
+	Explanation string     `json:"explanation"`
+	Edits       []FileEdit `json:"edits"`
+}
+
+type Diagnostic struct {
+	Path    string `json:"path"`
+	Message string `json:"message"`
+	Source  string `json:"source,omitempty"`
+	Code    string `json:"code,omitempty"`
+}
+
+type FixRequest struct {
+	Prompt      string       `json:"prompt,omitempty"`
+	Files       []FileInput  `json:"files"`
+	Diagnostics []Diagnostic `json:"diagnostics,omitempty"`
+	TestOutput  string       `json:"test_output,omitempty"`
+}
+
+type FixResponse = EditResponse
+
+type Agent struct {
+	client *ai.Client
+	search websearch.Searcher
+}
+
+func New() (*Agent, error) {
+	client, err := ai.NewClient()
+	if err != nil {
+		return nil, err
+	}
+
+	searcher := websearch.NewFromEnv()
+	return &Agent{client: client, search: searcher}, nil
+}
+
+func (a *Agent) Client() *ai.Client {
+	return a.client
+}
+
+func (a *Agent) Chat(ctx context.Context, req ChatRequest) (ChatResponse, error) {
+	var webResults []WebResult
+	if req.WebSearch.Enabled && a.search != nil {
+		query := strings.TrimSpace(req.WebSearch.Query)
+		if query == "" {
+			// fallback: use last user content
+			for i := len(req.Messages) - 1; i >= 0; i-- {
+				if req.Messages[i].Role == "user" {
+					query = req.Messages[i].Content
+					break
+				}
+			}
+		}
+		if query != "" {
+			results, err := a.search.Search(ctx, query)
+			if err == nil {
+				webResults = make([]WebResult, 0, len(results))
+				for _, r := range results {
+					webResults = append(webResults, WebResult{Title: r.Title, URL: r.URL, Snippet: r.Snippet})
+				}
+			}
+		}
+	}
+
+	messages := make([]ai.Message, 0, len(req.Messages)+1)
+	if len(webResults) > 0 {
+		messages = append(messages, ai.Message{Role: "system", Content: "Web search results (use as optional context):\n" + formatWebResults(webResults)})
+	}
+	messages = append(messages, req.Messages...)
+
+	if len(req.Attach) > 0 {
+		images, err := convertAttachmentsToImages(req.Attach)
+		if err != nil {
+			return ChatResponse{}, err
+		}
+		text, err := a.client.ChatWithImages(ctx, messages, images)
+		if err != nil {
+			return ChatResponse{}, err
+		}
+		return ChatResponse{Assistant: text, WebResults: webResults}, nil
+	}
+
+	text, err := a.client.ChatWithContext(ctx, messages)
+	if err != nil {
+		return ChatResponse{}, err
+	}
+	return ChatResponse{Assistant: text, WebResults: webResults}, nil
+}
+
+func (a *Agent) Edit(ctx context.Context, req EditRequest) (EditResponse, error) {
+	if strings.TrimSpace(req.Prompt) == "" {
+		return EditResponse{}, fmt.Errorf("prompt is required")
+	}
+	if len(req.Files) == 0 {
+		return EditResponse{}, fmt.Errorf("files is required")
+	}
+
+	var webContext string
+	if req.WebSearch.Enabled && a.search != nil {
+		query := strings.TrimSpace(req.WebSearch.Query)
+		if query == "" {
+			query = req.Prompt
+		}
+		if query != "" {
+			results, err := a.search.Search(ctx, query)
+			if err == nil {
+				webContext = formatWebResultsSimple(results)
+			}
+		}
+	}
+
+	system := "You are an AI coding agent. Return ONLY a JSON object with shape: {\"explanation\":string,\"edits\":[{\"path\":string,\"new_content\":string}]} .\n" +
+		"Rules: preserve unrelated code; keep formatting; ensure code compiles; if you can't, explain in explanation and return the original content."
+	if req.Instructions != "" {
+		system += "\nExtra instructions: " + req.Instructions
+	}
+	if webContext != "" {
+		system += "\n\nWeb context:\n" + webContext
+	}
+
+	var b strings.Builder
+	for _, f := range req.Files {
+		cleanPath := filepath.ToSlash(f.Path)
+		b.WriteString("FILE: " + cleanPath + "\n")
+		b.WriteString("```\n")
+		b.WriteString(f.Content)
+		if !strings.HasSuffix(f.Content, "\n") {
+			b.WriteString("\n")
+		}
+		b.WriteString("```\n\n")
+	}
+
+	messages := []ai.Message{
+		{Role: "system", Content: system},
+		{Role: "user", Content: "Task: " + req.Prompt + "\n\n" + b.String()},
+	}
+
+	text, err := a.client.ChatWithContext(ctx, messages)
+	if err != nil {
+		return EditResponse{}, err
+	}
+
+	resp, err := parseEditResponse(text)
+	if err != nil {
+		return EditResponse{}, err
+	}
+	return resp, nil
+}
+
+func (a *Agent) Fix(ctx context.Context, req FixRequest) (FixResponse, error) {
+	if len(req.Files) == 0 {
+		return FixResponse{}, fmt.Errorf("files is required")
+	}
+
+	prompt := strings.TrimSpace(req.Prompt)
+	if prompt == "" {
+		prompt = "Fix the reported problems and make the code compile/tests pass."
+	}
+
+	system := "You are an AI coding agent fixing code. Return ONLY JSON: {\"explanation\":string,\"edits\":[{\"path\":string,\"new_content\":string}]}. " +
+		"Keep changes minimal and targeted; do not refactor unrelated code."
+
+	var b strings.Builder
+	if len(req.Diagnostics) > 0 {
+		b.WriteString("DIAGNOSTICS:\n")
+		for _, d := range req.Diagnostics {
+			b.WriteString("- " + filepath.ToSlash(d.Path) + ": " + d.Message)
+			if d.Source != "" {
+				b.WriteString(" (" + d.Source + ")")
+			}
+			b.WriteString("\n")
+		}
+		b.WriteString("\n")
+	}
+	if req.TestOutput != "" {
+		b.WriteString("TEST OUTPUT:\n")
+		b.WriteString(req.TestOutput)
+		b.WriteString("\n\n")
+	}
+
+	for _, f := range req.Files {
+		cleanPath := filepath.ToSlash(f.Path)
+		b.WriteString("FILE: " + cleanPath + "\n")
+		b.WriteString("```\n")
+		b.WriteString(f.Content)
+		if !strings.HasSuffix(f.Content, "\n") {
+			b.WriteString("\n")
+		}
+		b.WriteString("```\n\n")
+	}
+
+	messages := []ai.Message{
+		{Role: "system", Content: system},
+		{Role: "user", Content: "Task: " + prompt + "\n\n" + b.String()},
+	}
+
+	text, err := a.client.ChatWithContext(ctx, messages)
+	if err != nil {
+		return FixResponse{}, err
+	}
+
+	resp, err := parseEditResponse(text)
+	if err != nil {
+		return FixResponse{}, err
+	}
+	return resp, nil
+}
+
+type imageInput struct {
+	Mime       string
+	DataBase64 string
+}
+
+func convertAttachmentsToImages(attachments []Attachment) ([]ai.ImageInput, error) {
+	images := make([]ai.ImageInput, 0, len(attachments))
+	for _, a := range attachments {
+		if a.Type != "image" {
+			continue
+		}
+		mime := a.Mime
+		if mime == "" {
+			mime = "image/png"
+		}
+		data := strings.TrimSpace(a.DataBase64)
+		if data == "" {
+			return nil, fmt.Errorf("image attachment missing data_base64")
+		}
+		// validate base64
+		if _, err := base64.StdEncoding.DecodeString(stripDataURLPrefix(data)); err != nil {
+			// might already be data url
+			if _, err2 := base64.StdEncoding.DecodeString(data); err2 != nil {
+				return nil, fmt.Errorf("invalid base64 image")
+			}
+		}
+		images = append(images, ai.ImageInput{Mime: mime, DataBase64: data})
+	}
+	if len(images) == 0 {
+		return nil, fmt.Errorf("no supported attachments")
+	}
+	return images, nil
+}
+
+func stripDataURLPrefix(s string) string {
+	if i := strings.Index(s, ","); i != -1 && strings.Contains(s[:i], "base64") {
+		return s[i+1:]
+	}
+	return s
+}
+
+func formatWebResults(results []WebResult) string {
+	var b strings.Builder
+	for i, r := range results {
+		b.WriteString(fmt.Sprintf("%d. %s\n   %s\n   %s\n", i+1, r.Title, r.URL, r.Snippet))
+	}
+	return b.String()
+}
+
+func formatWebResultsSimple(results []websearch.Result) string {
+	var b strings.Builder
+	for i, r := range results {
+		b.WriteString(fmt.Sprintf("%d. %s\n   %s\n   %s\n", i+1, r.Title, r.URL, r.Snippet))
+	}
+	return b.String()
+}
+
+func mustGetEnv(key, fallback string) string {
+	if v := os.Getenv(key); v != "" {
+		return v
+	}
+	return fallback
+}

package/internal/agent/parse.go

@@ -0,0 +1,25 @@
+package agent
+
+import (
+	"encoding/json"
+	"fmt"
+	"strings"
+)
+
+func parseEditResponse(text string) (EditResponse, error) {
+	trimmed := strings.TrimSpace(text)
+	// Some models wrap JSON in markdown fences
+	trimmed = strings.TrimPrefix(trimmed, "```json")
+	trimmed = strings.TrimPrefix(trimmed, "```")
+	trimmed = strings.TrimSuffix(trimmed, "```")
+	trimmed = strings.TrimSpace(trimmed)
+
+	var resp EditResponse
+	if err := json.Unmarshal([]byte(trimmed), &resp); err != nil {
+		return EditResponse{}, fmt.Errorf("model did not return valid JSON: %w", err)
+	}
+	if len(resp.Edits) == 0 {
+		return EditResponse{}, fmt.Errorf("model returned no edits")
+	}
+	return resp, nil
+}

package/internal/agents/base.go

@@ -0,0 +1,154 @@
+package agents
+
+import (
+	"context"
+	"fmt"
+	
+	"github.com/ddhanush1/dcode/internal/registry"
+)
+
+// BaseAgent provides common functionality for all agents
+type BaseAgent struct {
+	definition *registry.AgentDefinition
+	history    []registry.Message
+	context    map[string]interface{}
+}
+
+// NewBaseAgent creates a new base agent
+func NewBaseAgent(def *registry.AgentDefinition) *BaseAgent {
+	return &BaseAgent{
+		definition: def,
+		history:    make([]registry.Message, 0),
+		context:    make(map[string]interface{}),
+	}
+}
+
+// AddMessage adds a message to history
+func (a *BaseAgent) AddMessage(role, content string, data map[string]interface{}) {
+	a.history = append(a.history, registry.Message{
+		Role:    role,
+		Content: content,
+		Data:    data,
+	})
+}
+
+// GetHistory returns conversation history
+func (a *BaseAgent) GetHistory() []registry.Message {
+	return a.history
+}
+
+// ClearHistory clears conversation history
+func (a *BaseAgent) ClearHistory() {
+	a.history = make([]registry.Message, 0)
+}
+
+// SetContext sets context data
+func (a *BaseAgent) SetContext(key string, value interface{}) {
+	a.context[key] = value
+}
+
+// GetContext gets context data
+func (a *BaseAgent) GetContext(key string) interface{} {
+	return a.context[key]
+}
+
+// GetDefinition returns the agent definition
+func (a *BaseAgent) GetDefinition() *registry.AgentDefinition {
+	return a.definition
+}
+
+// AgentManager manages agent lifecycle and execution
+type AgentManager struct {
+	registry      *registry.AgentRegistry
+	toolRegistry  *registry.ToolRegistry
+	currentAgent  string
+	activeAgents  map[string]*BaseAgent
+}
+
+// NewAgentManager creates a new agent manager
+func NewAgentManager(agentRegistry *registry.AgentRegistry, toolRegistry *registry.ToolRegistry) *AgentManager {
+	return &AgentManager{
+		registry:     agentRegistry,
+		toolRegistry: toolRegistry,
+		currentAgent: "generalist",
+		activeAgents: make(map[string]*BaseAgent),
+	}
+}
+
+// SwitchAgent switches to a different agent
+func (m *AgentManager) SwitchAgent(agentID string) error {
+	if !m.registry.Has(agentID) {
+		return fmt.Errorf("agent not found: %s", agentID)
+	}
+	
+	m.currentAgent = agentID
+	return nil
+}
+
+// GetCurrentAgent returns the current agent
+func (m *AgentManager) GetCurrentAgent() (*registry.AgentDefinition, error) {
+	return m.registry.Get(m.currentAgent)
+}
+
+// Execute executes the current agent with input
+func (m *AgentManager) Execute(ctx context.Context, input *registry.AgentInput) (*registry.AgentOutput, error) {
+	agent, err := m.GetCurrentAgent()
+	if err != nil {
+		return nil, err
+	}
+	
+	// Get or create active agent instance
+	activeAgent, ok := m.activeAgents[agent.ID]
+	if !ok {
+		activeAgent = NewBaseAgent(agent)
+		m.activeAgents[agent.ID] = activeAgent
+	}
+	
+	// Add to history only if query is not empty
+	if input.Query != "" {
+		activeAgent.AddMessage("user", input.Query, nil)
+	}
+	
+	// Execute agent
+	output, err := agent.Executor(input)
+	if err != nil {
+		return nil, err
+	}
+	
+	// Add response to history only if message is not empty
+	if output.Message != "" {
+		activeAgent.AddMessage("assistant", output.Message, output.Data)
+	}
+	
+	return output, nil
+}
+
+// GetHistory returns history for an agent
+func (m *AgentManager) GetHistory(agentID string) []registry.Message {
+	if agent, ok := m.activeAgents[agentID]; ok {
+		return agent.GetHistory()
+	}
+	return []registry.Message{}
+}
+
+// ClearHistory clears history for an agent
+func (m *AgentManager) ClearHistory(agentID string) {
+	if agent, ok := m.activeAgents[agentID]; ok {
+		agent.ClearHistory()
+	}
+}
+
+// AddMessage adds a message to an agent's history
+func (m *AgentManager) AddMessage(agentID, role, content string, data map[string]interface{}) {
+	agent, ok := m.activeAgents[agentID]
+	if !ok {
+		// Create if doesn't exist
+		agentDef, err := m.registry.Get(agentID)
+		if err != nil {
+			return
+		}
+		agent = NewBaseAgent(agentDef)
+		m.activeAgents[agentID] = agent
+	}
+	agent.AddMessage(role, content, data)
+}

package/internal/agents/documenter.go

@@ -0,0 +1,77 @@
+package agents
+
+import (
+	"github.com/ddhanush1/dcode/internal/ai"
+	"github.com/ddhanush1/dcode/internal/registry"
+)
+
+const documenterSystemPrompt = `You are a Documentation Specialist, an AI agent specialized in creating clear, comprehensive documentation.
+
+Your documentation skills:
+- README files and project overviews
+- API documentation (REST, GraphQL, gRPC)
+- Code comments and inline documentation
+- Architecture documentation
+- User guides and tutorials
+- Contributing guidelines
+- Changelog management
+
+When writing documentation:
+1. Start with a clear purpose and audience
+2. Use clear, concise language
+3. Include practical examples
+4. Structure information logically
+5. Keep it up-to-date with code changes
+6. Use appropriate formatting (Markdown, etc.)
+
+Documentation types:
+- **README**: Project overview, setup, usage, examples
+- **API Docs**: Endpoints, parameters, responses, examples
+- **Code Comments**: Explain WHY, not WHAT
+- **Architecture**: System design, component relationships
+- **Guides**: Step-by-step tutorials for common tasks
+- **Reference**: Comprehensive API/function reference
+
+Best practices:
+- Write for your audience's level
+- Include code examples that actually work
+- Use diagrams where helpful
+- Keep documentation close to code
+- Update docs when code changes
+- Make it searchable and navigable
+
+Use available tools to:
+- read_file: Understand existing code/docs
+- write_file: Create/update documentation
+- glob: Find files to document
+- git_log: Understand project evolution
+- grep: Search for patterns
+- ls: Explore project structure
+
+Create documentation that helps people understand and use the code effectively.`
+
+// GetDocumenterTools returns tools best suited for documentation
+func GetDocumenterTools() []string {
+	return []string{
+		"read_file",
+		"write_file",
+		"glob",
+		"ls",
+		"git_log",
+		"grep",
+		"save_memory",
+		"recall_memory",
+	}
+}
+
+// RegisterDocumenterAgent registers the documenter agent
+func RegisterDocumenterAgent(agentRegistry *registry.AgentRegistry, aiClient *ai.Client) error {
+	return agentRegistry.Register(&registry.AgentDefinition{
+		ID:           "documenter",
+		Name:         "Documentation Specialist",
+		Description:  "Specialized in creating clear, comprehensive documentation",
+		SystemPrompt: documenterSystemPrompt,
+		Tools:        GetDocumenterTools(),
+		Executor:     CreateGeneralistExecutor(aiClient),
+	})
+}