fix: Restqdrant
Browse files- utils/llm_caller.py +1 -1
- utils/youtube_extractor.py +1 -1
utils/llm_caller.py
CHANGED
@@ -32,7 +32,7 @@ class LLMCaller:
|
|
32 |
self.embedding_model = SentenceTransformer("BAAI/bge-m3")
|
33 |
self.collection_name = "demo_bge_m3"
|
34 |
|
35 |
-
async def basic_query(self, user_prompt: str, max_tokens: int =
|
36 |
|
37 |
try:
|
38 |
completion = self.client.chat.completions.create(
|
|
|
32 |
self.embedding_model = SentenceTransformer("BAAI/bge-m3")
|
33 |
self.collection_name = "demo_bge_m3"
|
34 |
|
35 |
+
async def basic_query(self, user_prompt: str, max_tokens: int = 2048, model: str = "aisingapore/Gemma-SEA-LION-v3-9B-IT") -> str:
|
36 |
|
37 |
try:
|
38 |
completion = self.client.chat.completions.create(
|
utils/youtube_extractor.py
CHANGED
@@ -7,7 +7,7 @@ class YoutubeExtractor:
|
|
7 |
|
8 |
def extract_transcript(self, video_id: str) -> Optional[List[Dict]]:
|
9 |
try:
|
10 |
-
transcript = self.ytt_api.
|
11 |
return transcript
|
12 |
except Exception as e:
|
13 |
print(f"An error occurred: {e}")
|
|
|
7 |
|
8 |
def extract_transcript(self, video_id: str) -> Optional[List[Dict]]:
|
9 |
try:
|
10 |
+
transcript = self.ytt_api.get_transcript(video_id,languages=['en', 'th'])
|
11 |
return transcript
|
12 |
except Exception as e:
|
13 |
print(f"An error occurred: {e}")
|