lee/graphify
1
0
Fork 0
mirror of https://github.com/safishamsi/graphify.git synced 2026-09-22 21:45:58 +00:00
Code Issues Packages Projects Releases Wiki Activity

Remove Anthropic API call from transcribe.py - agent generates Whisper prompt itself

Browse Source 
Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
This commit is contained in:
Safi
2026-04-10 15:55:22 +01:00
co-authored by Claude Sonnet 4.6
parent db07e9f93a
commit 2a7d2f4fdc
11 changed files with 146 additions and 191 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_transcribe.py
+4 -32
View File
@@ -1,13 +1,10 @@
"""Tests for graphify.transcribe — video/audio transcription support."""
from __future__ import annotations
import json
import os
from pathlib import Path
from unittest.mock import MagicMock, patch
import sys
import pytest
from graphify.transcribe import (
@@ -47,38 +44,13 @@ def test_build_whisper_prompt_env_override(monkeypatch):
assert prompt == "Custom domain hint."
def test_build_whisper_prompt_llm_success():
"""Successful LLM call returns generated prompt with punctuation suffix."""
def test_build_whisper_prompt_returns_topic_string():
"""Returns a topic-based prompt from god node labels — no LLM call."""
god_nodes = [{"label": "neural networks"}, {"label": "transformers"}, {"label": "attention"}]
fake_response = MagicMock()
fake_response.content = [MagicMock(text="Machine learning and deep learning research")]
mock_anthropic = MagicMock()
mock_anthropic.Anthropic.return_value.messages.create.return_value = fake_response
with patch.dict(os.environ, {}, clear=False):
os.environ.pop("GRAPHIFY_WHISPER_PROMPT", None)
with patch.dict(sys.modules, {"anthropic": mock_anthropic}):
prompt = build_whisper_prompt(god_nodes)
assert "Machine learning" in prompt
assert "punctuation" in prompt.lower()
def test_build_whisper_prompt_llm_failure_fallback():
"""If LLM call raises, falls back to topic-based prompt."""
god_nodes = [{"label": "kubernetes"}, {"label": "docker"}, {"label": "helm"}]
mock_anthropic = MagicMock()
mock_anthropic.Anthropic.return_value.messages.create.side_effect = Exception("API error")
with patch.dict(os.environ, {}, clear=False):
os.environ.pop("GRAPHIFY_WHISPER_PROMPT", None)
with patch.dict(sys.modules, {"anthropic": mock_anthropic}):
prompt = build_whisper_prompt(god_nodes)
assert "kubernetes" in prompt.lower() or "docker" in prompt.lower()
prompt = build_whisper_prompt(god_nodes)
assert "neural networks" in prompt.lower() or "transformers" in prompt.lower()
assert "punctuation" in prompt.lower()