Spaces:

Subhajit01
/

SmartLead

Sleeping

SmartLead / src /services /llm_client.py

Subhajit Chakraborty

add imp files

723bbe6 about 1 year ago

505 Bytes

from langchain_google_genai import ChatGoogleGenerativeAI
from langchain.chains import LLMChain
from dotenv import load_dotenv
import os
load_dotenv()

class LLMClient:
    def __init__(self):
        self.client = ChatGoogleGenerativeAI(model="gemini-2.5-flash", google_api_key = os.getenv('GOOGLE_API_KEY'))
        # self.client.invoke()
        
    async def generate_text(self, prompt: str) -> str:
        response = await self.client.invoke([prompt]).generations[0][0].text
        return response

	from langchain_google_genai import ChatGoogleGenerativeAI
	from langchain.chains import LLMChain
	from dotenv import load_dotenv
	import os
	load_dotenv()

	class LLMClient:
	def __init__(self):
	self.client = ChatGoogleGenerativeAI(model="gemini-2.5-flash", google_api_key = os.getenv('GOOGLE_API_KEY'))
	# self.client.invoke()

	async def generate_text(self, prompt: str) -> str:
	response = await self.client.invoke([prompt]).generations[0][0].text
	return response