mirror of
https://github.com/microsoft/graphrag.git
synced 2025-07-07 17:11:12 +00:00
23 lines
794 B
Python
23 lines
794 B
Python
# Copyright (c) 2024 Microsoft Corporation.
|
|
# Licensed under the MIT License
|
|
from datashaper import TableContainer, VerbInput
|
|
|
|
|
|
def str_append(
|
|
input: VerbInput, source_column: str, target_column: str, string_to_append: str
|
|
):
|
|
"""A custom verb that appends a string to a column"""
|
|
# by convention, we typically use "column" as the input column name and "to" as the output column name, but you can use whatever you want
|
|
# just as long as the "args" in the workflow reference match the function signature
|
|
input_data = input.get_input()
|
|
output_df = input_data.copy()
|
|
output_df[target_column] = output_df[source_column].apply(
|
|
lambda x: f"{x}{string_to_append}"
|
|
)
|
|
return TableContainer(table=output_df)
|
|
|
|
|
|
custom_verbs = {
|
|
"str_append": str_append,
|
|
}
|