summaryrefslogtreecommitdiffstats
path: root/forefront/typing.py
diff options
context:
space:
mode:
authort.me/xtekky <98614666+xtekky@users.noreply.github.com>2023-04-27 11:34:05 +0200
committert.me/xtekky <98614666+xtekky@users.noreply.github.com>2023-04-27 11:34:05 +0200
commit98d2b4109e8cd06ffd46ce9fe747bf889a4e7fff (patch)
tree2ad018b50660a2e6ec397a71f07c6c937071fc62 /forefront/typing.py
parentUpdate main.py (diff)
downloadgpt4free-98d2b4109e8cd06ffd46ce9fe747bf889a4e7fff.tar
gpt4free-98d2b4109e8cd06ffd46ce9fe747bf889a4e7fff.tar.gz
gpt4free-98d2b4109e8cd06ffd46ce9fe747bf889a4e7fff.tar.bz2
gpt4free-98d2b4109e8cd06ffd46ce9fe747bf889a4e7fff.tar.lz
gpt4free-98d2b4109e8cd06ffd46ce9fe747bf889a4e7fff.tar.xz
gpt4free-98d2b4109e8cd06ffd46ce9fe747bf889a4e7fff.tar.zst
gpt4free-98d2b4109e8cd06ffd46ce9fe747bf889a4e7fff.zip
Diffstat (limited to '')
-rw-r--r--forefront/typing.py37
1 files changed, 37 insertions, 0 deletions
diff --git a/forefront/typing.py b/forefront/typing.py
new file mode 100644
index 00000000..0fff6b18
--- /dev/null
+++ b/forefront/typing.py
@@ -0,0 +1,37 @@
+class ForeFrontResponse:
+ 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