fixes

gpt3discord.py

@@ -33,7 +33,7 @@ from services.environment_service import EnvService
 from models.openai_model import Model
 
 
-__version__ = "11.1.1"
+__version__ = "11.1.5"
 
 
 PID_FILE = Path("bot.pid")

requirements.txt

@@ -24,4 +24,6 @@ openai-whisper
 unidecode==1.3.6
 tqdm==4.64.1
 docx2txt==0.8
-pytest-asyncio==0.21.0
+pytest-asyncio==0.21.0
+aiohttp~=3.8.4
+pytest~=7.2.2

tests/test_requests.py

@@ -1,14 +1,13 @@
 from pathlib import Path
 
 import pytest
-from models.openai_model import Models, Model
+from models.openai_model import Model
 from transformers import GPT2TokenizerFast
-import asyncio
 
 from services.usage_service import UsageService
-import os
 
-# Non-ChatGPT
+
+# Non-ChatGPT -> TODO: make generic test and loop through text models
 @pytest.mark.asyncio
 async def test_send_req():
 
@@ -28,4 +27,25 @@ async def test_send_req_gpt():
     model = Model(usage_service)
     prompt = 'how many hours are in a day?'
     res = await model.send_request(prompt, None, is_chatgpt_request=True, model="gpt-3.5-turbo")
-    assert '24' in res['choices'][0]['message']['content']
+    assert '24' in res['choices'][0]['message']['content']
+
+
+# GPT4 version
+@pytest.mark.asyncio
+async def test_send_req_gpt4():
+    usage_service = UsageService(Path("../tests"))
+    model = Model(usage_service)
+    prompt = 'how many hours are in a day?'
+    res = await model.send_request(prompt, None, is_chatgpt_request=True, model="gpt-4")
+    assert '24' in res['choices'][0]['message']['content']
+
+
+# Edit request -> currently broken due to endpoint
+# @pytest.mark.asyncio
+# async def test_send_edit_req():
+#     usage_service = UsageService(Path("../tests"))
+#     model = Model(usage_service)
+#     text = 'how many hours are in a day?'
+#     res = await model.send_edit_request(text, codex=True)
+#     assert '24' in res['choices'][0]['text']
+