DocDoeAI / scripts /debug_sarvam_class.py
asnannp's picture
Deploy backend cd4237ff: support routes + rate limit + exam_date nullable + upload 413 fix
7c6ffa6
Raw
History Blame Contribute Delete
1.52 kB
import sys
import os
from pathlib import Path
# Add backend to sys.path
backend_dir = Path("C:/Users/USER/OneDrive/Desktop/STARTUP/backend")
sys.path.append(str(backend_dir))
from app.services.ai_provider import SarvamAIProvider
from app.core.config import get_settings
from pydantic import BaseModel
class SimpleSchema(BaseModel):
greeting: str
fact: str
def test_sarvam_real():
settings = get_settings()
print(f"AI_PROVIDER: {settings.ai_provider}")
print(f"SARVAM_MODEL_MAIN: {settings.sarvam_model_main}")
provider = SarvamAIProvider()
try:
result = provider.generate_notes(
context="Photosynthesis is the process by which green plants and some other organisms use sunlight to synthesize foods from carbon dioxide and water. Photosynthesis in plants generally involves the green pigment chlorophyll and generates oxygen as a byproduct.",
language="English",
mode="explain_simple",
metadata={"subject": "Biology", "chapter": "Plant Physiology"}
)
print("\nSUCCESS!")
print(f"Model used: {provider.model_name}")
print(f"Is fallback: {provider.is_fallback}")
print(f"Result keys: {list(result.keys())}")
print(f"Title: {result.get('title')}")
print(f"Summary: {result.get('student_level_summary')}")
except Exception as e:
print(f"\nFAILED: {e}")
import traceback
traceback.print_exc()
if __name__ == "__main__":
test_sarvam_real()