Merge pull request #20 from workflowai/guillaume/new-run-endpoint

Use new run endpoint

Author: guillaq

.github/workflows/e2e.yml

@@ -9,7 +9,7 @@ on:
   workflow_dispatch:
 
 env:
-  WORKFLOWAI_TEST_API_URL: https://api.workflowai.dev
+  WORKFLOWAI_TEST_API_URL: https://run.workflowai.dev
 
 jobs:
   e2e-tests:

.vscode/settings.json

@@ -1,49 +1,43 @@
 {
-    "[python]": {
-        "editor.codeActionsOnSave": {
-            "source.fixAll": "explicit",
-            "source.organizeImports": "explicit"
-        },
-        "editor.defaultFormatter": "charliermarsh.ruff",
-    },
-    "python.analysis.typeCheckingMode": "strict",
-    "python.analysis.ignore": [
-        "**/*_pb2.py",
-        "**/*_pb2.pyi",
-        "**/*_pb2_grpc.py",
-        "*.proto"
-    ],
-    "ruff.lint.args": [
-        "--extend-ignore=E501",
-        "--extend-exclude=/**/3rdparty_repositories/**/*.py",
-        "--extend-exclude=/**/site-packages/**/*.py",
-    ],
-    "python.testing.unittestEnabled": false,
-    "python.testing.pytestEnabled": true,
-    "python.testing.pytestArgs": [
-        "-vv",
-    ],
-    "python.languageServer": "Pylance",
-    "python.analysis.autoImportCompletions": true,
-    "editor.suggestSelection": "first",
-    "editor.quickSuggestions": {
-        "other": true,
-        "comments": false,
-        "strings": false
-    },
-    "editor.acceptSuggestionOnEnter": "on",
-    "editor.tabCompletion": "on",
-    "python.analysis.completeFunctionParens": true,
-    "files.exclude": {
-        "**/.git": true,
-        "**/.svn": true,
-        "**/.hg": true,
-        "**/CVS": true,
-        "**/.DS_Store": true,
-        "**/Thumbs.db": true,
-        "ai_model_cache": true,
-        ".mypy_cache": true,
-        ".venv": true,
-        ".ruff_cache": true,
+  "[python]": {
+    "editor.codeActionsOnSave": {
+      "source.fixAll": "explicit",
+      "source.organizeImports": "explicit"
     },
+    "editor.defaultFormatter": "charliermarsh.ruff"
+  },
+  "python.analysis.typeCheckingMode": "strict",
+  "python.analysis.ignore": [
+    "**/*_pb2.py",
+    "**/*_pb2.pyi",
+    "**/*_pb2_grpc.py",
+    "*.proto"
+  ],
+
+  "python.testing.unittestEnabled": false,
+  "python.testing.pytestEnabled": true,
+  "python.testing.pytestArgs": ["-vv"],
+  "python.languageServer": "Pylance",
+  "python.analysis.autoImportCompletions": true,
+  "editor.suggestSelection": "first",
+  "editor.quickSuggestions": {
+    "other": true,
+    "comments": false,
+    "strings": false
+  },
+  "editor.acceptSuggestionOnEnter": "on",
+  "editor.tabCompletion": "on",
+  "python.analysis.completeFunctionParens": true,
+  "files.exclude": {
+    "**/.git": true,
+    "**/.svn": true,
+    "**/.hg": true,
+    "**/CVS": true,
+    "**/.DS_Store": true,
+    "**/Thumbs.db": true,
+    "ai_model_cache": true,
+    ".mypy_cache": true,
+    ".venv": true,
+    ".ruff_cache": true
+  }
 }

conftest.py

@@ -2,7 +2,7 @@
 from freezegun import freeze_time
 
 
-@pytest.fixture()
+@pytest.fixture
 def frozen_time():
     with freeze_time("2024-01-01T00:00:00Z") as frozen_time:
         yield frozen_time

examples/city_to_capital_task.py

@@ -1,6 +1,6 @@
-from pydantic import BaseModel, Field
+from pydantic import BaseModel, Field  # pyright: ignore [reportUnknownVariableType]
 
-from workflowai import Task, TaskVersionReference
+from workflowai import Task, VersionReference
 
 
 class CityToCapitalTaskInput(BaseModel):
@@ -12,7 +12,8 @@ class CityToCapitalTaskInput(BaseModel):
 
 class CityToCapitalTaskOutput(BaseModel):
     capital: str = Field(
-        description="The capital of the specified city", examples=["Tokyo"],
+        description="The capital of the specified city",
+        examples=["Tokyo"],
     )
 
 
@@ -22,6 +23,4 @@ class CityToCapitalTask(Task[CityToCapitalTaskInput, CityToCapitalTaskOutput]):
     input_class: type[CityToCapitalTaskInput] = CityToCapitalTaskInput
     output_class: type[CityToCapitalTaskOutput] = CityToCapitalTaskOutput
 
-    version: TaskVersionReference = TaskVersionReference(
-        iteration=4,
-    )
+    version: VersionReference = 4