Merge pull request #13 from YeetCode-devs/staging/prathamdby

Override the default author for workflow commits, implement a new test, and use the default model on all tests

.github/workflows/add-license.yml

@@ -26,5 +26,7 @@ jobs:
 
     - uses: GuillaumeFalourd/[email protected]
       with:
+          email: "[email protected]"
+          name: "GitHub Actions Bot"
           commit_message: "Add license to files"
           target_branch: main

tests/test_bing.py

@@ -16,20 +16,22 @@
 
 from g4f.client import Client
 from g4f.Provider import Bing
+from g4f.models import default
 
 
 def generate_response() -> str:
     client = Client(provider=Bing)
 
     try:
         response = client.chat.completions.create(
-            model="gpt-4.0-turbo",
+            model=default,
             messages=[
                 {"role": "user", "content": "Say hi, with your response starting with START and ending with END"}
             ],
         )
     except:
         print("ERROR: Could not create a prompt!")
+        raise Exception("ERROR: Could not create a prompt!")
 
     return response.choices[0].message.content
 

tests/test_freechatgpt.py

@@ -0,0 +1,32 @@
+from g4f.client import Client
+from g4f.Provider import FreeChatgpt
+from g4f.models import default
+
+
+def generate_response() -> str:
+    client = Client(provider=FreeChatgpt)
+
+    try:
+        response = client.chat.completions.create(
+            model=default,
+            messages=[
+                {"role": "user", "content": "Say hi, with your response starting with START and ending with END"}
+            ],
+        )
+    except:
+        print("ERROR: Could not create a prompt!")
+        raise Exception("ERROR: Could not create a prompt!")
+
+    return response.choices[0].message.content
+
+
+class TestOutput:
+    def test_output(self):
+        response = generate_response()
+
+        if len(response) > 0:
+            print("✅ FreeChatgpt is up!")
+        else:
+            print("❌ FreeChatgpt is down...")
+
+        assert response.startswith("START") and response.endswith("END")