From 9ac43cb34f5f660ef61472ffc8ad2b59179f69a7 Mon Sep 17 00:00:00 2001 From: Samay Date: Mon, 13 Jul 2026 20:37:27 +0530 Subject: [PATCH] fix: updated ai engine dependencies and switched to groq for cloud --- ai_engine/local_pipeline.py | 6 +++--- ai_engine/requirements.txt | 5 +++++ 2 files changed, 8 insertions(+), 3 deletions(-) diff --git a/ai_engine/local_pipeline.py b/ai_engine/local_pipeline.py index a966690..8d685bb 100644 --- a/ai_engine/local_pipeline.py +++ b/ai_engine/local_pipeline.py @@ -7,7 +7,7 @@ from typing import TypedDict, List, Dict, Any from faster_whisper import WhisperModel -from langchain_ollama import ChatOllama +from langchain_groq import ChatGroq from langchain_core.messages import SystemMessage, HumanMessage from langgraph.graph import StateGraph, START, END @@ -305,10 +305,10 @@ def node_transcribe(state: VideoState) -> VideoState: return {"audio_path": audio, "video_duration": dur, "words": words, "segments": sentences, "status": "Transcribed"} def node_plan(state: VideoState) -> VideoState: - print("\n--- [Ollama] Strategizing Top 0.1% Cinematic B-Roll Placements ---") + print("\n--- [Groq] Strategizing Top 0.1% Cinematic B-Roll Placements ---") segments = state["segments"] - llm = ChatOllama(model="qwen3:4b", temperature=0.2) + llm = ChatGroq(model="llama3-70b-8192", temperature=0.2, groq_api_key=os.getenv("GROQ_API_KEY")) transcript_summary = "" for i, seg in enumerate(segments): diff --git a/ai_engine/requirements.txt b/ai_engine/requirements.txt index 8615f9a..5c10188 100644 --- a/ai_engine/requirements.txt +++ b/ai_engine/requirements.txt @@ -15,3 +15,8 @@ google-generativeai==0.8.2 # YouTube Download yt-dlp>=2024.1.0 +# AI Pipeline (Cloud-ready) +faster_whisper>=1.0.3 +langgraph>=0.1.5 +langchain_groq>=0.1.6 +langchain_core>=0.2.11