taspol commited on
Commit
18036d7
·
1 Parent(s): 3f8b197

fix: Restqdrant

Browse files
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 = 1024, model: str = "aisingapore/Gemma-SEA-LION-v3-9B-IT") -> str:
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.fetch(video_id,languages=['en', 'th'])
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}")