This commit is contained in:
Alexander Myasoedov
2025-03-10 13:25:17 +02:00
parent 1226e2059d
commit 91c99e642f
2 changed files with 10 additions and 7 deletions
+8 -6
View File
@@ -1,7 +1,7 @@
import base64 import base64
from enum import Enum from enum import Enum
from urllib.parse import urlparse from urllib.parse import urlparse
import httpx import httpx
from pydantic import BaseModel from pydantic import BaseModel
@@ -161,11 +161,13 @@ def parse_http_spec(http_spec: str) -> LLMSpec:
method, url = lines[0].split(" ")[0:2] method, url = lines[0].split(" ")[0:2]
# Check url validity # Check url validity
valid_url= urlparse(url) valid_url = urlparse(url)
# if missing the correct formatting ://, urlparse.netloc will be empty # if missing the correct formatting ://, urlparse.netloc will be empty
if valid_url.scheme not in ("http", "https") or not valid_url.netloc: if valid_url.scheme not in ("http", "https") or not valid_url.netloc:
raise InvalidHTTPSpecError(f"Invalid URL: {url}. Ensure it starts with 'http://' or 'https://'") raise InvalidHTTPSpecError(
f"Invalid URL: {url}. Ensure it starts with 'http://' or 'https://'"
)
# Initialize headers and body # Initialize headers and body
headers = {} headers = {}
body = "" body = ""
@@ -121,7 +121,8 @@ class CloudRLPromptSelector(PromptSelectionInterface):
current_prompt: str, current_prompt: str,
reward: float, reward: float,
passed_guard: bool, passed_guard: bool,
) -> None: ... ) -> None:
...
class QLearningPromptSelector(PromptSelectionInterface): class QLearningPromptSelector(PromptSelectionInterface):