Skip to content

Commit b579c62

Browse files
committed
Clear cache before pickle
1 parent 7e27784 commit b579c62

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

src/guidellm/backend/openai.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -127,6 +127,13 @@ def __init__(
127127
self._async_client: Optional[httpx.AsyncClient] = None
128128
self._request_template_str = settings.openai.request_template
129129

130+
def __getstate__(self) -> object:
131+
state = self.__dict__.copy()
132+
# Templates are not serializable
133+
# so we delete it before pickling
134+
state.pop("request_template", None)
135+
return state
136+
130137
@cached_property
131138
def request_template(self) -> jinja2.Template:
132139
j2_env = jinja2.Environment(loader=jinja2.BaseLoader(), autoescape=True)
@@ -163,6 +170,7 @@ def info(self) -> dict[str, Any]:
163170
"project": self.project,
164171
"text_completions_path": TEXT_COMPLETIONS_PATH,
165172
"chat_completions_path": CHAT_COMPLETIONS_PATH,
173+
"request_template": self._request_template_str,
166174
}
167175

168176
async def check_setup(self):

0 commit comments

Comments
 (0)