Serve the agent to a terminal client

[onechat] / main.go

diff --git a/main.go b/main.go
index 9cd619a3cf8a1e207cfecfbbd331321b9fbe5977..680c58d9a1c3884b5f8bab7bc8ef04d3874ee12c 100644 (file)
--- a/main.go
+++ b/main.go
@@ -33,6 +33,21 @@ func run() error {
                if err != nil {
                        return err
                }
                if err != nil {
                        return err
                }

+       case len(args) == 1 && args[0] == "daemon":
+               err := daemon(ctx, os.Stdout)
+               if err != nil {
+                       return err
+               }
+       case len(args) == 1 && args[0] == "chat":
+               err := chat(ctx, os.Stdin, os.Stdout)
+               if err != nil {
+                       return err
+               }
+       case len(args) > 0 && args[0] == "probe":
+               err := probe(ctx, args[1:], os.Stdout)
+               if err != nil {
+                       return err
+               }

        default:
                return fmt.Errorf("unknown command %q", strings.Join(args, " "))
        }
        default:
                return fmt.Errorf("unknown command %q", strings.Join(args, " "))
        }