mirror of
https://github.com/langbot-app/LangBot.git
synced 2026-06-12 16:56:02 +00:00
feat: session基本功能及单测
This commit is contained in:
16
tests/test_session_console.py
Normal file
16
tests/test_session_console.py
Normal 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))
|
||||
Reference in New Issue
Block a user