Merge pull request #7 from replicate/cog-runtime-cog-server

Remaining code renaming

README.md

@@ -1,18 +1,17 @@
-Coggo
-=====
+# Cog Runtime
 
 Alt-core [Cog] runtime implementation.
 
 The original [Cog] seeks to be a great developer tool and also an arguably great
-production runtime. Coggo is focused on being a fantastic production runtime _only_.
+production runtime. Cog runtime is focused on being a fantastic production runtime _only_.
 
-How is Coggo formed?
+How is `cog-runtime` formed?
 
 ```mermaid
 sequenceDiagram
     participant r8
-    participant server as coggo-server
-    participant runner as cog.internal.file_runner
+    participant server as cog-server
+    participant runner as coglet
     participant predictor as Predictor
     r8->>server: Boot
     activate server
@@ -38,15 +37,15 @@ sequenceDiagram
 ```
 
 This sequence is simplified, but the rough idea is that the Replicate platform (`r8`)
-depends on `coggo-server` to provide an HTTP API in front of a `cog.internal.file_runner`
+depends on `cog-server` to provide an HTTP API in front of a `coglet`
 that communicates via files and signals.
 
-## `coggo-server`
+## `cog-server`
 
 Go-based HTTP server that known how to spawn and communicate with
-`cog.internal.file_runner`.
+`coglet`.
 
-## `cog.internal.file_runner`
+## `coglet`
 
 Python-based model runner with zero dependencies outside of the standard library.
 The same in-process API provided by [Cog] is avaaliable, e.g.:
@@ -62,7 +61,7 @@ class MyPredictor(BasePredictor):
 In addition to simple cases like the above, the runner is async by default and supports
 continuous batching.
 
-Communication with the `coggo-server` parent process is managed via input and output files
+Communication with the `cog-server` parent process is managed via input and output files
 and the following signals:
 
 - `SIGUSR1` model is ready

TODO.md

@@ -1,6 +1,6 @@
-* coggo-server: Output file encoding and upload
-* coggo-server: Webhook interval
-* coggo-server: E2E tests
-* coggo-server: Publish go binary
-* python: Publish python source tree
+* cog-server: Output file encoding and upload
+* cog-server: Webhook interval
+* cog-server: E2E tests
+* cog-server: Publish go binary
+* coglet: Publish python source tree
 * Install both in monobase build.sh

cmd/coggo-server/main.go → cmd/cog-server/main.go

@@ -17,11 +17,11 @@ import (
 	"github.com/replicate/go/version"
 	_ "go.uber.org/automaxprocs"
 
-	"github.com/replicate/coggo/internal/server"
-	"github.com/replicate/coggo/internal/util"
+	"github.com/replicate/cog-runtime/internal/server"
+	"github.com/replicate/cog-runtime/internal/util"
 )
 
-var logger = logging.New("coggo-server")
+var logger = logging.New("cog-server")
 
 type Config struct {
 	Host                  string `ff:"long: host, default: 0.0.0.0, usage: HTTP server host"`
@@ -43,12 +43,12 @@ func main() {
 	log := logger.Sugar()
 
 	var cfg Config
-	flags := ff.NewFlagSet("coggo-server")
+	flags := ff.NewFlagSet("cog-server")
 	must.Do(flags.AddStruct(&cfg))
 
 	cmd := &ff.Command{
-		Name:  "coggo-server",
-		Usage: "coggo-server [FLAGS]",
+		Name:  "cog-server",
+		Usage: "cog-server [FLAGS]",
 		Flags: flags,
 		Exec: func(ctx context.Context, args []string) error {
 			if err := cfg.Validate(); err != nil {
@@ -67,7 +67,7 @@ func main() {
 			}
 			workingDir := cfg.WorkingDir
 			if workingDir == "" {
-				workingDir = must.Get(os.MkdirTemp("", "coggo-server-"))
+				workingDir = must.Get(os.MkdirTemp("", "cog-server-"))
 			}
 			log.Infow("configuration", "working-dir", workingDir, "module-name", moduleName, "class-name", className)
 

go.mod

@@ -1,4 +1,4 @@
-module github.com/replicate/coggo
+module github.com/replicate/cog-runtime
 
 go 1.23.3
 

internal/server/runner.go

@@ -18,7 +18,7 @@ import (
 
 	"github.com/replicate/go/must"
 
-	"github.com/replicate/coggo/internal/util"
+	"github.com/replicate/cog-runtime/internal/util"
 )
 
 var LOG_REGEX = regexp.MustCompile(`^\[pid=(?P<pid>[^\\]+)] (?P<msg>.*)$`)

script/coggo.sh → script/cog-server.sh

@@ -18,7 +18,7 @@ export PATH="$base_dir/python/.venv/bin:$PATH"
 export PYTHONPATH="$base_dir/python"
 rm -rf tmp
 mkdir -p tmp
-go run cmd/coggo-server/main.go \
+go run cmd/cog-server/main.go \
     --working-dir tmp \
     --module-name "tests.runners.$module" \
     --class-name Predictor \