INNER CODE UNIT · Python

fetch_ai_response

AnsahMohammad/shots-studio · legacy/web/core.py:123

    def fetch_ai_response(self, request_data, headers):
        url = f"{self.base_url}/{self.model_name}:generateContent?key={self.api_key}"

        try:
            response = requests.post(
                url, headers=headers, json=request_data, timeout=self.timeout
            )
            response_json = response.json()

            ai_response_text = (
                response_json.get("candidates", [{}])[0]
                .get("content", {})
                .get("parts", [{}])[0]
                .get("text", None)
            )

            if not ai_response_text:
                return "No response from AI", response.status_code

View source record →