From 1d9e995a0dcdfe6f2f763476ed725ece58831011 Mon Sep 17 00:00:00 2001 From: Krish Dholakia Date: Tue, 8 Aug 2023 11:19:28 -0700 Subject: [PATCH 1/2] adding support for anthropic, azure, cohere, llama2 --- hacker_news/news.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/hacker_news/news.py b/hacker_news/news.py index 6a3aa09..5bb3349 100644 --- a/hacker_news/news.py +++ b/hacker_news/news.py @@ -4,7 +4,8 @@ import re import time from json import JSONDecodeError - +import litellm +from litellm import completion import openai from slugify import slugify from summarizer import Summarizer @@ -193,7 +194,7 @@ def openai_complete(self, prompt, need_json): ) answer = resp['choices'][0]['text'].strip() else: - resp = openai.ChatCompletion.create( + resp = completion( messages=[ {'role': 'user', 'content': prompt}, ], From f85235e9941a305e754d697dfa606a2c7e1dbc5e Mon Sep 17 00:00:00 2001 From: Krish Dholakia Date: Fri, 11 Aug 2023 14:36:53 -0700 Subject: [PATCH 2/2] Update requirements.txt --- requirements.txt | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/requirements.txt b/requirements.txt index f8f175b..a8ec620 100644 --- a/requirements.txt +++ b/requirements.txt @@ -21,4 +21,5 @@ python-dotenv==1.0.0 python_slugify==8.0.1 sqlalchemy==2.0.19 psycopg2==2.9.6 -humanize==4.7.0 \ No newline at end of file +humanize==4.7.0 +litellm==0.1.381