feat: session基本功能及单测

This commit is contained in:
Rock Chin
2022-12-07 22:50:49 +08:00
parent 4303487d32
commit 908cb1634b
3 changed files with 20 additions and 1 deletions

View File

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