vodorod/Dorod-Sky
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Set organization_id for posthog through event (#277)

Browse Source
This commit is contained in:
Kerem Yilmaz
2024-05-08 01:33:06 -07:00
committed by GitHub
parent cc91c1b2b6
commit 42d652f381
2 changed files with 7 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
skyvern/forge/agent.py
View File

@@ -629,7 +629,9 @@ class ForgeAgent:
)
prompt_template = "extract-action"
if app.EXPERIMENTATION_PROVIDER.is_feature_enabled_cached(
"USE_CLAUDE3_SONNET", task.workflow_run_id or task.task_id
"USE_CLAUDE3_SONNET",
task.workflow_run_id or task.task_id,
properties={"organization_id": task.organization_id},
):
LOG.info("Using Claude3 Sonnet prompt template for action extraction")
prompt_template = "extract-action-claude3-sonnet"

8
skyvern/forge/sdk/experimentation/providers.py
View File

@@ -10,17 +10,17 @@ class BaseExperimentationProvider(ABC):
result_map: dict[str, dict[str, bool]] = {}
@abstractmethod
def is_feature_enabled(self, feature_name: str, distinct_id: str) -> bool:
def is_feature_enabled(self, feature_name: str, distinct_id: str, properties: dict | None = None) -> bool:
"""Check if a specific feature is enabled."""
def is_feature_enabled_cached(self, feature_name: str, distinct_id: str) -> bool:
def is_feature_enabled_cached(self, feature_name: str, distinct_id: str, properties: dict | None = None) -> bool:
if feature_name not in self.result_map:
self.result_map[feature_name] = {}
if distinct_id not in self.result_map[feature_name]:
self.result_map[feature_name][distinct_id] = self.is_feature_enabled(feature_name, distinct_id)
self.result_map[feature_name][distinct_id] = self.is_feature_enabled(feature_name, distinct_id, properties)
return self.result_map[feature_name][distinct_id]
class NoOpExperimentationProvider(BaseExperimentationProvider):
def is_feature_enabled(self, feature_name: str, distinct_id: str) -> bool:
def is_feature_enabled(self, feature_name: str, distinct_id: str, properties: dict | None = None) -> bool:
return False