KeyError: 'data' with stella_embeddings function which calls the Landing AI embedding API

[aswaryhcl]

When I am using the VisionAgentV2 or VisionAgentCoderV2, I am get this error after sometime:

response = session.post(url, data=payload).json()
data = response["data"]  # <-- KeyError Probably occurs here

"C:\ProgramData\miniconda3\envs\vision-agent-env\lib\site-packages\vision_agent\agent\vision_agent_v2.py", line 227, in __call__ return self.chat(input_msg)[-1].content File "C:\ProgramData\miniconda3\envs\vision-agent-env\lib\site-packages\vision_agent\agent\vision_agent_v2.py", line 274, in chat updated_chat = maybe_run_action( File "C:\ProgramData\miniconda3\envs\vision-agent-env\lib\site-packages\vision_agent\agent\vision_agent_v2.py", line 112, in maybe_run_action context = coder.generate_code(extracted_chat, code_interpreter=code_interpreter) File "C:\ProgramData\miniconda3\envs\vision-agent-env\lib\site-packages\vision_agent\agent\vision_agent_coder_v2.py", line 417, in generate_code code_context = self.generate_code_from_plan( File "C:\ProgramData\miniconda3\envs\vision-agent-env\lib\site-packages\vision_agent\agent\vision_agent_coder_v2.py", line 467, in generate_code_from_plan tool_docs = retrieve_tools(plan_context.instructions, self.tool_recommender) File "C:\ProgramData\miniconda3\envs\vision-agent-env\lib\site-packages\vision_agent\agent\vision_agent_coder_v2.py", line 57, in retrieve_tools tools = tool_recommender.top_k(inst, k=1, thresh=0.3) File "C:\ProgramData\miniconda3\envs\vision-agent-env\lib\site-packages\vision_agent\sim\sim.py", line 176, in top_k embedding = get_embedding( File "C:\ProgramData\miniconda3\envs\vision-agent-env\lib\site-packages\vision_agent\sim\sim.py", line 33, in get_embedding return emb_call([text]) File "C:\ProgramData\miniconda3\envs\vision-agent-env\lib\site-packages\vision_agent\sim\sim.py", line 285, in emb_call return stella_embeddings(text)[0] # type: ignore File "C:\ProgramData\miniconda3\envs\vision-agent-env\lib\site-packages\vision_agent\sim\sim.py", line 74, in stella_embeddings data = response["data"] KeyError: 'data'