NeoAnd/adk-python
1
0
Fork 0
mirror of https://github.com/EvolutionAPI/adk-python.git synced 2025-07-14 09:51:25 -06:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fixes: eval_result endpoint throws due to folder not existing.

Browse Source 
PiperOrigin-RevId: 758508672
This commit is contained in:
Wei Sun (Jack) 2025-05-13 22:02:56 -07:00 committed by Copybara-Service
parent b33bdb9a03
commit 71bd808a14
1 changed files with 11 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
src/google/adk/cli/fast_api.py
View File

@ -12,6 +12,7 @@
# See the License for the specific language governing permissions and # See the License for the specific language governing permissions and
# limitations under the License. # limitations under the License.
import asyncio import asyncio
from contextlib import asynccontextmanager from contextlib import asynccontextmanager
import importlib import importlib
@ -20,7 +21,6 @@ import json
import logging import logging
import os import os
from pathlib import Path from pathlib import Path
import re
import sys import sys
import time import time
import traceback import traceback
@ -48,12 +48,9 @@ from opentelemetry.exporter.cloud_trace import CloudTraceSpanExporter
from opentelemetry.sdk.trace import export from opentelemetry.sdk.trace import export
from opentelemetry.sdk.trace import ReadableSpan from opentelemetry.sdk.trace import ReadableSpan
from opentelemetry.sdk.trace import TracerProvider from opentelemetry.sdk.trace import TracerProvider
from opentelemetry.sdk.trace.export.in_memory_span_exporter import InMemorySpanExporter
from pydantic import alias_generators
from pydantic import BaseModel
from pydantic import ConfigDict
from pydantic import ValidationError from pydantic import ValidationError
from starlette.types import Lifespan from starlette.types import Lifespan
from typing_extensions import override
from ..agents import RunConfig from ..agents import RunConfig
from ..agents.base_agent import BaseAgent from ..agents.base_agent import BaseAgent
@ -121,6 +118,7 @@ class InMemoryExporter(export.SpanExporter):
self._spans = [] self._spans = []
self.trace_dict = trace_dict self.trace_dict = trace_dict
@override
def export( def export(
self, spans: typing.Sequence[ReadableSpan] self, spans: typing.Sequence[ReadableSpan]
) -> export.SpanExportResult: ) -> export.SpanExportResult:
@ -137,15 +135,16 @@ class InMemoryExporter(export.SpanExporter):
self._spans.extend(spans) self._spans.extend(spans)
return export.SpanExportResult.SUCCESS return export.SpanExportResult.SUCCESS
@override
def force_flush(self, timeout_millis: int = 30000) -> bool:
return True
def get_finished_spans(self, session_id: str): def get_finished_spans(self, session_id: str):
trace_ids = self.trace_dict.get(session_id, None) trace_ids = self.trace_dict.get(session_id, None)
if trace_ids is None or not trace_ids: if trace_ids is None or not trace_ids:
return [] return []
return [x for x in self._spans if x.context.trace_id in trace_ids] return [x for x in self._spans if x.context.trace_id in trace_ids]
def force_flush(self, timeout_millis: int = 30000) -> bool:
return True
def clear(self): def clear(self):
self._spans.clear() self._spans.clear()
@ -600,6 +599,10 @@ def get_fast_api_app(
app_eval_history_directory = os.path.join( app_eval_history_directory = os.path.join(
agent_dir, app_name, ".adk", "eval_history" agent_dir, app_name, ".adk", "eval_history"
) )
if not os.path.exists(app_eval_history_directory):
return []
eval_result_files = [ eval_result_files = [
file.removesuffix(_EVAL_SET_RESULT_FILE_EXTENSION) file.removesuffix(_EVAL_SET_RESULT_FILE_EXTENSION)
for file in os.listdir(app_eval_history_directory) for file in os.listdir(app_eval_history_directory)