Internal change

nijat12 · copybara-github · commit 55a2ffc2c773 · 2025-07-02T06:31:54.000-07:00
GitOrigin-RevId: ec84ac38a1c7b5a2f9be1d6bf317cc12e99f0ce8
diff --git a/library/evals/autorating/hallucination_autorater.py b/library/evals/autorating/hallucination_autorater.py
@@ -26,7 +26,7 @@
     format_comments,
     format_summary,
 )
-from vertex_model import VertexModel, run_tasks_in_parallel
+from models.vertex_model import VertexModel, run_tasks_in_parallel
 
 
 class HallucinationAutorater:
diff --git a/library/evals/autorating/hallucination_autorater_test.py b/library/evals/autorating/hallucination_autorater_test.py
@@ -19,7 +19,7 @@
 import pytest
 from autorating_utils import EvalInput
 from hallucination_autorater import HallucinationAutorater
-from vertex_model import VertexModel
+from models.vertex_model import VertexModel
 
 
 @pytest.fixture
diff --git a/library/evals/autorating/run_autoraters.py b/library/evals/autorating/run_autoraters.py
@@ -38,7 +38,7 @@
 
 from hallucination_autorater import HallucinationAutorater
 from autorating_utils import read_csv
-from vertex_model import VertexModel
+from models.vertex_model import VertexModel
 
 
 async def main():
diff --git a/library/requirements.txt b/library/requirements.txt
@@ -0,0 +1,5 @@
+pytest==8.4.1
+pytest-asyncio==1.0.0
+pandas==2.3.0
+vertexai==1.71.1
+google-genai==1.23.0
diff --git a/models/__init__.py b/models/__init__.py
diff --git a/models/vertex_model.py b/models/vertex_model.py

Original file line number	Diff line number	Diff line change
`@@ -26,7 +26,7 @@`
`26`	`26`	`format_comments,`
`27`	`27`	`format_summary,`
`28`	`28`	`)`
`29`		`-from vertex_model import VertexModel, run_tasks_in_parallel`
	`29`	`+from models.vertex_model import VertexModel, run_tasks_in_parallel`
`30`	`30`
`31`	`31`
`32`	`32`	`class HallucinationAutorater:`