124 lines
2.9 KiB
Go
124 lines
2.9 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"flag"
|
|
"fmt"
|
|
"os"
|
|
|
|
"ollie/internal/agent"
|
|
|
|
"ollie/internal/backend"
|
|
"ollie/internal/config"
|
|
execpkg "ollie/internal/exec"
|
|
"ollie/internal/tui"
|
|
"ollie/pkg/core"
|
|
)
|
|
|
|
func main() {
|
|
sessionFlag := flag.String("session", "", "resume a session by ID")
|
|
promptFlag := flag.String("prompt", "", "run a single prompt non-interactively and exit")
|
|
flag.Parse()
|
|
extraArgs := flag.Args()
|
|
|
|
home, _ := os.UserHomeDir()
|
|
agentsDir := home + "/.config/ollie/agents"
|
|
sessionsDir := home + "/.config/ollie/sessions"
|
|
if err := os.MkdirAll(sessionsDir, 0700); err != nil {
|
|
fmt.Fprintln(os.Stderr, "sessions dir:", err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
be, err := backend.New()
|
|
if err != nil {
|
|
fmt.Fprintln(os.Stderr, "failed to create backend:", err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
backendName := agent.ResolveBackendName()
|
|
modelName := os.Getenv("OLLIE_MODEL")
|
|
if modelName == "" {
|
|
modelName = agent.DefaultModelForBackend(backendName)
|
|
}
|
|
|
|
builtinExec := execpkg.New(
|
|
home+"/.local/state/ollie",
|
|
home+"/.cache/ollie/exec",
|
|
)
|
|
|
|
agentName := os.Getenv("OLLIE_AGENT")
|
|
if agentName == "" {
|
|
agentName = "default"
|
|
}
|
|
|
|
sessionID := agent.NewSessionID()
|
|
var resumeMessages []backend.Message
|
|
if *sessionFlag != "" {
|
|
sessionPath := sessionsDir + "/" + *sessionFlag + ".json"
|
|
data, readErr := os.ReadFile(sessionPath)
|
|
if readErr != nil {
|
|
fmt.Fprintln(os.Stderr, "--session:", readErr)
|
|
os.Exit(1)
|
|
}
|
|
var ps agent.PersistedSession
|
|
if jsonErr := json.Unmarshal(data, &ps); jsonErr != nil {
|
|
fmt.Fprintln(os.Stderr, "--session: bad JSON:", jsonErr)
|
|
os.Exit(1)
|
|
}
|
|
sessionID = ps.ID
|
|
resumeMessages = ps.Messages
|
|
if ps.Agent != "" && len(extraArgs) == 0 {
|
|
agentName = ps.Agent
|
|
}
|
|
}
|
|
if len(extraArgs) > 0 {
|
|
agentName = extraArgs[0]
|
|
}
|
|
|
|
cfgPath := agent.AgentConfigPath(agentsDir, agentName)
|
|
cfg, cfgErr := config.Load(cfgPath)
|
|
env := agent.BuildAgentEnv(cfg, builtinExec)
|
|
|
|
var initialSession *agent.Session
|
|
if len(resumeMessages) > 0 {
|
|
initialSession = agent.RestoreSession(resumeMessages, agent.ContextConfig{
|
|
FixedOverheadChars: env.CtxOverhead,
|
|
})
|
|
}
|
|
|
|
agentCore := agent.NewAgentCore(agent.AgentCoreConfig{
|
|
Backend: be,
|
|
BackendName: backendName,
|
|
ModelName: modelName,
|
|
AgentName: agentName,
|
|
AgentsDir: agentsDir,
|
|
SessionsDir: sessionsDir,
|
|
SessionID: sessionID,
|
|
Session: initialSession,
|
|
Env: env,
|
|
BuiltinExec: builtinExec,
|
|
})
|
|
|
|
if cfgErr != nil {
|
|
fmt.Fprintln(os.Stderr, "agent config:", cfgErr)
|
|
}
|
|
for _, msg := range env.Messages {
|
|
fmt.Fprintln(os.Stderr, msg)
|
|
}
|
|
if len(resumeMessages) > 0 {
|
|
fmt.Fprintf(os.Stderr, "session: %s (resumed)\n", sessionID)
|
|
} else {
|
|
fmt.Fprintf(os.Stderr, "session: %s\n", sessionID)
|
|
}
|
|
|
|
env.Hooks.Run(core.HookAgentSpawn)
|
|
|
|
if *promptFlag != "" {
|
|
agentCore.Submit(context.Background(), *promptFlag, tui.MakeOutputFn(os.Stdout))
|
|
return
|
|
}
|
|
|
|
tui.New(agentCore).Run(context.Background())
|
|
}
|