diff --git a/example_segmentation.py b/segment_search_queries.py.py similarity index 97% rename from example_segmentation.py rename to segment_search_queries.py.py index fb44d01..e99213d 100644 --- a/example_segmentation.py +++ b/segment_search_queries.py.py @@ -58,7 +58,7 @@ def segment(data: str) -> MultiSearch: model="gpt-3.5-turbo-0613", temperature=0, functions=[MultiSearch.openai_schema], - function_call={"name": MultiSearch.openai_schema['name']}, + function_call={"name": MultiSearch.openai_schema["name"]}, messages=[ { "role": "system",