Skip to content

Commit

Permalink
cp
Browse files Browse the repository at this point in the history
  • Loading branch information
schrockn committed Dec 24, 2024
1 parent d0c18e9 commit cc6fae1
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 5 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@

import click

from dagster_components.core.component import ComponentMetadata, ComponentTypeRegistry
from dagster_components.core.component import ComponentTypeMetadata, ComponentTypeRegistry
from dagster_components.core.deployment import (
CodeLocationProjectContext,
find_enclosing_code_location_root_path,
Expand Down Expand Up @@ -41,7 +41,7 @@ def list_component_types_command(ctx: click.Context) -> None:
output: Dict[str, Any] = {}
for key, component_type in context.list_component_types():
package, name = key.rsplit(".", 1)
output[key] = ComponentMetadata(
output[key] = ComponentTypeMetadata(
name=name,
package=package,
**component_type.get_metadata(),
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -66,7 +66,7 @@ def build_defs(self, context: "ComponentLoadContext") -> Definitions: ...
def load(cls, context: "ComponentLoadContext") -> Self: ...

@classmethod
def get_metadata(cls) -> "ComponentInternalMetadata":
def get_metadata(cls) -> "ComponentTypeInternalMetadata":
docstring = cls.__doc__
clean_docstring = _clean_docstring(docstring) if docstring else None

Expand Down Expand Up @@ -104,14 +104,14 @@ def _get_click_cli_help(command: click.Command) -> str:
return formatter.getvalue()


class ComponentInternalMetadata(TypedDict):
class ComponentTypeInternalMetadata(TypedDict):
summary: Optional[str]
description: Optional[str]
generate_params_schema: Optional[Any] # json schema
component_params_schema: Optional[Any] # json schema


class ComponentMetadata(ComponentInternalMetadata):
class ComponentTypeMetadata(ComponentTypeInternalMetadata):
name: str
package: str

Expand Down

0 comments on commit cc6fae1

Please sign in to comment.