Skip to content

Commit c16bf4a

Browse files
committed
fix issue
1 parent 9863ec7 commit c16bf4a

2 files changed

Lines changed: 5 additions & 6 deletions

File tree

.github/workflows/dagger-fraud-sim.yml

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -22,6 +22,4 @@ jobs:
2222
GROQ_API_KEY: ${{ secrets.GROQ_API_KEY }}
2323
run: |
2424
dagger run --cloud authentication-service/auth-service:fraud-v1 \
25-
call run-fraud-sim \
26-
--gemini-api-key env:GEMINI_API_KEY \
27-
--groq-api-key env:GROQ_API_KEY
25+
call run-fraud-sim

dagger/src/fraud_simulation/main.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,15 +1,16 @@
11
"""Dagger module for fraud simulation."""
22
import dagger
3-
from dagger import dag, function, object_type
3+
from dagger import dag, function, object_type, Doc
4+
from typing import Annotated
45

56
@object_type
67
class FraudSimulation:
78
"""A Dagger module to run fraud simulation pipelines."""
89
@function
910
async def run_fraud_sim(
1011
self,
11-
gemini_api_key: dagger.Secret,
12-
groq_api_key: dagger.Secret,
12+
gemini_api_key: Annotated[dagger.Secret, Doc("Gemini API token")],
13+
groq_api_key: Annotated[dagger.Secret, Doc("Groq API token")],
1314
) -> str:
1415
"""
1516
Runs the fraud simulation pipeline with the provided AI API keys.

0 commit comments

Comments
 (0)