temp commit

test_enrich.py

@@ -0,0 +1,42 @@
+import asyncio
+from sqlalchemy.ext.asyncio import create_async_engine, async_sessionmaker
+from app.settings import settings
+from app.db.models import ScholarProfile
+from app.services.domains.ingestion.application import ScholarIngestionService
+from app.services.domains.scholar.parser_types import PublicationCandidate
+from app.services.domains.openalex.client import OpenAlexClient
+
+async def main():
+    service = ScholarIngestionService()
+    scholar = ScholarProfile(
+        id=1,
+        scholar_id="SaiiI5MAAAAJ",
+        name="Test Scholar"
+    )
+    pubs = [
+        PublicationCandidate(
+            title="A fast quantum mechanical algorithm for database search",
+            year=1996,
+            citation_count=1000,
+            authors_text="LK Grover",
+            venue_text="Proceedings of the 28th Annual ACM Symposium on the Theory of Computing",
+            cluster_id=None,
+            title_url=None,
+            pdf_url=None
+        )
+    ]
+    
+    print("Enriching...")
+    try:
+        enriched = await service._enrich_publications_with_openalex(scholar, pubs)
+        for p in enriched:
+            print(f"Title: {p.title}")
+            print(f"Year: {p.year}")
+            print(f"Citations: {p.citation_count}")
+            print(f"PDF URL: {p.pdf_url}")
+    except Exception as e:
+        import traceback
+        traceback.print_exc()
+
+if __name__ == "__main__":
+    asyncio.run(main())