How to support Rest API calling in custom wrapper asynchronous model invoke function.

[sriramraja-krish]

Hey I am building an custom wrapper for Model invoke logic, So in my case I am using rest api, I am changing the below code into rest api, async def a_generate(
self, prompt: str, schema: Optional[BaseModel] = None
) -> str:
"""Asynchronously generates text from a prompt.

    Args:
        prompt: Text prompt
        schema: Optional Pydantic model for structured output

    Returns:
        Generated text response or structured output as Pydantic model
    """
    if schema is not None:
        response = await self.client.aio.models.generate_content(
            model=self.model_name,
            contents=prompt,
            config=types.GenerateContentConfig(
                response_mime_type="application/json",
                response_schema=schema,
                safety_settings=self.model_safety_settings,
                temperature=self.temperature,
            ),
        )

I am trying above function to change as below,
async with aiohttp.ClientSession() as session:
async with session.post(url, headers=headers, json=payload) as resp:
if resp.status != 200:
raise Exception(f"Request failed with status {resp.status}")
data = await resp.json()

But the problem is the prompt has not include previous opinions , So the request passed without context, I tried asyn call for http request still no use , when i try run_asynx =False , It works for first metric, but not for other metric , how to achieve this ,