Function create_llm_chain_google

src/llm.py:84–132

(prompt_template: str, llm, history: Optional[str] = None)

    return generation

def create_llm_chain_google(prompt_template: str, llm, history: Optional[str] = None) -> str:
    url = "https://generativelanguage.googleapis.com/v1beta/models/gemini-1.5-flash:generateContent"
    
    headers = {
        "Content-Type": "application/json"
    }
    
    # If history exists, include it in the prompt
    full_prompt = f"{history}\n{prompt_template}, you reply in json file" if history else prompt_template
    
    data = {
        "contents": [{
            "parts": [{"text": full_prompt}]
        }]
    }
    
    params = {
        "key": "your google key"
    }
    
    try:
        response = requests.post(url, headers=headers, json=data, params=params)
        response.raise_for_status()
        
        json_response = response.json()
        
        # Extract the text from candidates[0].content.parts[0].text
        if (json_response.get("candidates") and 
            len(json_response["candidates"]) > 0 and 
            json_response["candidates"][0].get("content") and 
            json_response["candidates"][0]["content"].get("parts") and 
            len(json_response["candidates"][0]["content"]["parts"]) > 0):
            
            output = json_response["candidates"][0]["content"]["parts"][0]["text"]
        else:
            raise ValueError("Unexpected response structure from Gemini API")
            
        output = str(output)
        output = output[7:-3]
        output = json.dumps({"output": output})
        logger("printing:")        
        logger(output)

        return output

    except requests.exceptions.RequestException as e:
        raise Exception(f"API request failed: {str(e)}")
    except ValueError as e:
        raise Exception(f"Error processing response: {str(e)}")