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
from enum import Enum
from urllib.parse import urlparse
import httpx
from pydantic import BaseModel
@@ -161,11 +161,13 @@ def parse_http_spec(http_spec: str) -> LLMSpec:
method, url = lines[0].split(" ")[0:2]
# Check url validity
valid_url= urlparse(url)
# if missing the correct formatting ://, urlparse.netloc will be empty
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://'")
valid_url = urlparse(url)
# if missing the correct formatting ://, urlparse.netloc will be empty
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://'"
)
# Initialize headers and body
headers = {}
body = ""
@@ -121,7 +121,8 @@ class CloudRLPromptSelector(PromptSelectionInterface):
current_prompt: str,
reward: float,
passed_guard: bool,
) -> None: ...
) -> None:
...
class QLearningPromptSelector(PromptSelectionInterface):