summaryrefslogtreecommitdiffstats
path: root/etc/tool/create_provider.py
diff options
context:
space:
mode:
Diffstat (limited to '')
-rw-r--r--etc/tool/create_provider.py8
1 files changed, 3 insertions, 5 deletions
diff --git a/etc/tool/create_provider.py b/etc/tool/create_provider.py
index abae8f96..361448e4 100644
--- a/etc/tool/create_provider.py
+++ b/etc/tool/create_provider.py
@@ -8,8 +8,7 @@ sys.path.append(str(Path(__file__).parent.parent.parent))
import g4f
def read_code(text):
- match = re.search(r"```(python|py|)\n(?P<code>[\S\s]+?)\n```", text)
- if match:
+ if match := re.search(r"```(python|py|)\n(?P<code>[\S\s]+?)\n```", text):
return match.group("code")
def input_command():
@@ -99,12 +98,11 @@ And replace "gpt-3.5-turbo" with `model`.
print()
response = "".join(response)
- code = read_code(response)
- if code:
+ if code := read_code(response):
with open(provider_path, "w") as file:
file.write(code)
print("Saved at:", provider_path)
- with open(f"g4f/Provider/__init__.py", "a") as file:
+ with open("g4f/Provider/__init__.py", "a") as file:
file.write(f"\nfrom .{name} import {name}")
else:
with open(provider_path, "r") as file: