Skip to content
New issue

Have a question about this project? # for a free GitHub account to open an issue and contact its maintainers and the community.

#

By clicking “#”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? # to your account

ENG-1110: Onboarding an Utility Model #334

Merged
merged 9 commits into from
Dec 10, 2024
Merged

ENG-1110: Onboarding an Utility Model #334

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
9 commits
Select commit Hold shift + click to select a range
789d2dc
Onboard Utilities
thiago-aixplain Nov 25, 2024
685db06
Tests for utility models
thiago-aixplain Nov 26, 2024
e5ff63d
Merge
thiago-aixplain Nov 26, 2024
d543a0e
Parse code function
thiago-aixplain Dec 3, 2024
66d8fd9
Adapting utility model onboarding to unit tests
thiago-aixplain Dec 4, 2024
9b3d3ba
Unit test for utility removal
thiago-aixplain Dec 4, 2024
989c9d0
Merge Dev
thiago-aixplain Dec 4, 2024
f830b59
Auto-describe utility model inputs
thiago-aixplain Dec 10, 2024
e7c3906
Merge branch 'development' into ENG-1110-onboardUtilities
thiago-aixplain Dec 10, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
12 changes: 9 additions & 3 deletions aixplain/modules/model/utils.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -120,13 +120,19 @@ def parse_code(code: Union[Text, Callable]) -> Tuple[Text, List, Text]:

if input_type in ["int", "float"]:
input_type = "number"
inputs.append(UtilityModelInput(name=input_name, type=DataType.NUMBER, description=""))
inputs.append(
UtilityModelInput(name=input_name, type=DataType.NUMBER, description=f"The {input_name} input is a number")
)
elif input_type == "bool":
input_type = "boolean"
inputs.append(UtilityModelInput(name=input_name, type=DataType.BOOLEAN, description=""))
inputs.append(
UtilityModelInput(name=input_name, type=DataType.BOOLEAN, description=f"The {input_name} input is a boolean")
)
elif input_type == "str":
input_type = "text"
inputs.append(UtilityModelInput(name=input_name, type=DataType.TEXT, description=""))
inputs.append(
UtilityModelInput(name=input_name, type=DataType.TEXT, description=f"The {input_name} input is a text")
)
else:
raise Exception(f"Utility Model Error: Unsupported input type: {input_type}")

Expand Down
12 changes: 8 additions & 4 deletions tests/unit/utility_test.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -24,7 +24,9 @@ def test_utility_model():
assert utility_model.name == "utility_model_test"
assert utility_model.description == "utility_model_test"
assert utility_model.code == "utility_model_test"
assert utility_model.inputs == [UtilityModelInput(name="originCode", description="", type=DataType.TEXT)]
assert utility_model.inputs == [
UtilityModelInput(name="originCode", description="The originCode input is a text", type=DataType.TEXT)
]
assert utility_model.output_examples == "output_description"


Expand Down Expand Up @@ -136,7 +138,9 @@ def test_parse_code():
with patch("aixplain.factories.file_factory.FileFactory.upload", return_value="code_link"):
code = "def main(originCode: str) -> str:\n return originCode"
code_link, inputs, description = parse_code(code)
assert inputs == [UtilityModelInput(name="originCode", description="", type=DataType.TEXT)]
assert inputs == [
UtilityModelInput(name="originCode", description="The originCode input is a text", type=DataType.TEXT)
]
assert description == ""
assert code_link == "code_link"

Expand All @@ -152,8 +156,8 @@ def main(a: int, b: int):
code = main
code_link, inputs, description = parse_code(code)
assert inputs == [
UtilityModelInput(name="a", description="", type=DataType.NUMBER),
UtilityModelInput(name="b", description="", type=DataType.NUMBER),
UtilityModelInput(name="a", description="The a input is a number", type=DataType.NUMBER),
UtilityModelInput(name="b", description="The b input is a number", type=DataType.NUMBER),
]
assert description == "This function adds two numbers"
assert code_link == "code_link"
Expand Down