summaryrefslogtreecommitdiffstats
path: root/ora/typing.py
diff options
context:
space:
mode:
authort.me/xtekky <98614666+xtekky@users.noreply.github.com>2023-04-01 15:08:23 +0200
committert.me/xtekky <98614666+xtekky@users.noreply.github.com>2023-04-01 15:08:23 +0200
commit493c37a81ecc34247189f7339632518268524655 (patch)
tree2d647b02b5c16ed5d0c2643112f4b7c31d1c62e3 /ora/typing.py
parentimportant note + new site (ora.sh) (diff)
downloadgpt4free-493c37a81ecc34247189f7339632518268524655.tar
gpt4free-493c37a81ecc34247189f7339632518268524655.tar.gz
gpt4free-493c37a81ecc34247189f7339632518268524655.tar.bz2
gpt4free-493c37a81ecc34247189f7339632518268524655.tar.lz
gpt4free-493c37a81ecc34247189f7339632518268524655.tar.xz
gpt4free-493c37a81ecc34247189f7339632518268524655.tar.zst
gpt4free-493c37a81ecc34247189f7339632518268524655.zip
Diffstat (limited to '')
-rw-r--r--ora/typing.py39
1 files changed, 39 insertions, 0 deletions
diff --git a/ora/typing.py b/ora/typing.py
new file mode 100644
index 00000000..f3f0aebf
--- /dev/null
+++ b/ora/typing.py
@@ -0,0 +1,39 @@
+class OraResponse:
+
+ class Completion:
+
+ class Choices:
+ def __init__(self, choice: dict) -> None:
+ self.text = choice['text']
+ self.content = self.text.encode()
+ self.index = choice['index']
+ self.logprobs = choice['logprobs']
+ self.finish_reason = choice['finish_reason']
+
+ def __repr__(self) -> str:
+ return f'''<__main__.APIResponse.Completion.Choices(\n text = {self.text.encode()},\n index = {self.index},\n logprobs = {self.logprobs},\n finish_reason = {self.finish_reason})object at 0x1337>'''
+
+ def __init__(self, choices: dict) -> None:
+ self.choices = [self.Choices(choice) for choice in choices]
+
+ class Usage:
+ def __init__(self, usage_dict: dict) -> None:
+ self.prompt_tokens = usage_dict['prompt_tokens']
+ self.completion_tokens = usage_dict['completion_tokens']
+ self.total_tokens = usage_dict['total_tokens']
+
+ def __repr__(self):
+ return f'''<__main__.APIResponse.Usage(\n prompt_tokens = {self.prompt_tokens},\n completion_tokens = {self.completion_tokens},\n total_tokens = {self.total_tokens})object at 0x1337>'''
+
+ def __init__(self, response_dict: dict) -> None:
+
+ self.response_dict = response_dict
+ self.id = response_dict['id']
+ self.object = response_dict['object']
+ self.created = response_dict['created']
+ self.model = response_dict['model']
+ self.completion = self.Completion(response_dict['choices'])
+ self.usage = self.Usage(response_dict['usage'])
+
+ def json(self) -> dict:
+ return self.response_dict \ No newline at end of file