Method WithSDK

sdk/python/runtime/main.go:390–439

Add the SDK package to the source directory This includes regenerating the client bindings for the current API schema (codegen).

(introspectionJSON *dagger.File)

// This includes regenerating the client bindings for the current API schema
// (codegen).
func (m *PythonSdk) WithSDK(introspectionJSON *dagger.File) *PythonSdk {
	if m.VendorPath != "" {
		src := m.SdkSourceDir
		// If not vendoring we don't care to remove this
		if m.Discovery.SdkHasFile("dist/") {
			src = src.WithoutDirectory("dist")
		}
		m.AddDirectory(m.VendorPath, src)
	}

	// Allow empty introspection to facilitate debugging the container with a
	// `dagger call module-runtime terminal` command.
	if introspectionJSON != nil {
		ctr := m.Container
		cmd := []string{"codegen"}

		// When not using the bundled codegen executable we can revert to executing directly
		if m.Discovery.SdkHasFile("dist/codegen") {
			ctr = ctr.
				WithMountedCache("/root/.shiv", dag.CacheVolume("shiv")).
				WithMountedFile("/usr/local/bin/codegen", m.SdkSourceDir.File("dist/codegen"))
		} else {
			ctr = ctr.
				WithWorkdir("/sdk").
				WithMountedDirectory("", m.SdkSourceDir)
			cmd = []string{
				"uv", "run", "--isolated", "--frozen", "--package", "codegen",
				"python", "-m", "codegen",
			}
		}

		genFile := ctr.
			// mounted schema as late as possible because it varies more often
			WithMountedFile(SchemaPath, introspectionJSON).
			WithExec(append(cmd, "generate", "-i", SchemaPath, "-o", "/gen.py")).
			File("/gen.py")

		genPath := UserGenPath

		// For now, patch vendored client library with generated bindings.
		// TODO: Always generate outside library, even if vendored.
		if m.VendorPath != "" {
			genPath = path.Join(m.VendorPath, SDKGenPath)
		}

		m.AddFile(genPath, genFile)
	}

	return m
}

// Add the module's source code
func (m *PythonSdk) WithSource() *PythonSdk {