[Bug fix] Fix test

[deshraj]

Description

Missed to fix the test in my last PR.

[codecov]

Codecov Report

Attention: 2 lines in your changes are missing coverage. Please review.

Comparison is base (d94aee8) 56.52% compared to head (44f0bd8) 56.54%.
Report is 2 commits behind head on main.

Files	Patch %	Lines
embedchain/llm/openai.py	66.66%	2 Missing ⚠️

Additional details and impacted files

@@            Coverage Diff             @@
##             main    #1255      +/-   ##
==========================================
+ Coverage   56.52%   56.54%   +0.01%     
==========================================
  Files         146      146              
  Lines        5935     5937       +2     
==========================================
+ Hits         3355     3357       +2     
  Misses       2580     2580              

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.