diff --git a/calvin/calvin.py b/calvin/calvin.py index c3161d8..8b867e6 100644 --- a/calvin/calvin.py +++ b/calvin/calvin.py @@ -1,9 +1,9 @@ import json import requests -# # # # -# Calvin - abstraction module over openai's API # -# # # # +# # # # +# Calvin - abstraction module over openai's API # +# # # # # Defile module-scoped variables @@ -13,7 +13,7 @@ # Initialize module -def initialize(_api_key, _engine): +def initialize(_api_key, _engine=engine): global engine, headers, prompts if _engine: @@ -86,8 +86,8 @@ def create_completion(payload): def complete_prompt(prompt): payload = { "prompt": prompt, - "max_tokens": 5, - "temperature": 1, + "max_tokens": 150, + "temperature": 0.9, "top_p": 1, "n": 1 } @@ -143,4 +143,4 @@ def test_create_completion(): "logprobs": null, "stop": "\n" } -""" \ No newline at end of file +"""