perf: 完善openai异常处理

This commit is contained in:
RockChinQ
2024-02-01 18:11:47 +08:00
parent a9d92115f8
commit 7026abe56a
14 changed files with 1195 additions and 9 deletions

View File

@@ -1,14 +1,14 @@
import config
import unittest
import pkg.openai.session
import pkg.openai.manager
import pkg.provider.session
import pkg.provider.manager
class TestOpenAISession(unittest.TestCase):
def test_session_console(self):
interact = pkg.openai.manager.OpenAIInteract(config.openai_config['api_key'], config.completion_api_params)
interact = pkg.provider.manager.OpenAIInteract(config.openai_config['api_key'], config.completion_api_params)
session = pkg.openai.session.Session('test')
session = pkg.provider.session.Session('test')
print(session.append('你好'))
print("#{}#".format(session.prompt))