| """ | |
| test_openviking.py | |
| ------------------ | |
| Verifies that the `tools.py` OpenViking and Gemini integrations | |
| can successfully retrieve L1/L2 summaries and texts. | |
| """ | |
| from src.agents.tools import IADC_SearchTool, VolveHistory_SearchTool | |
| def run_tests(): | |
| print("Initializing Tests...") | |
| iadc_tool = IADC_SearchTool() | |
| volve_tool = VolveHistory_SearchTool() | |
| # Test 1: IADC Definition Search | |
| print("\\n--- Test 1 (IADC Definition) ---") | |
| res1 = iadc_tool._run("What is non-productive time (NPT)?") | |
| print("Result snippet:", res1[:500]) | |
| # Test 2: Volve Historical Search | |
| print("\\n--- Test 2 (Volve Event) ---") | |
| res2 = volve_tool._run("Did any stuck pipe incidents occur on 15/9-19 A?") | |
| print("Result snippet:", res2[:500]) | |
| if __name__ == "__main__": | |
| run_tests() | |