File tree Expand file tree Collapse file tree 1 file changed +8
-0
lines changed
Expand file tree Collapse file tree 1 file changed +8
-0
lines changed Original file line number Diff line number Diff 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 ):
You can’t perform that action at this time.
0 commit comments