thisisdev commited on
Commit
65b8f24
·
verified ·
1 Parent(s): 6ec34fa

Update Helper

Browse files
Files changed (1) hide show
  1. langchain_helper.py +32 -0
langchain_helper.py CHANGED
@@ -9,3 +9,35 @@ from secret_key import openapi_key
9
  # Reading the OPEN API KEY
10
  os.environ['OPENAI_API_KEY'] = openapi_key
11
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
9
  # Reading the OPEN API KEY
10
  os.environ['OPENAI_API_KEY'] = openapi_key
11
 
12
+ # First chain of command: Name of the organization
13
+ def generate_name(feild):
14
+ prompt_template = PromptTemplate(
15
+ input_variables=["feild"],
16
+ template="I want to open a edtech organization for {feild} domain. Suggest a great name for this and the course structure to be followed.",
17
+ )
18
+
19
+ name_chain = LLMChain(llm = model, prompt = prompt_template, output_key = "organization_name")
20
+
21
+ # Second chain of command: Tips
22
+
23
+ prompt_template = PromptTemplate(
24
+ input_variables=["specifics"],
25
+ template="Suggest me tips of how we can elevate the {specifics} for generative AI, and return it in comma seperated format",
26
+ )
27
+
28
+ specific_chain = LLMChain(llm = model, prompt = prompt_template, output_key = "tips")
29
+
30
+
31
+ chain = SequentialChain(
32
+ chains = [name_chain, specific_chain],
33
+ input_variables = ["feild", "specifics"], # Added "specifics" to the input variables
34
+ output_variables = ["organization_name", "tips"]
35
+ )
36
+
37
+ resp = chain({"feild" : feild})
38
+
39
+ return resp
40
+
41
+
42
+ if __name__ == "__main__":
43
+ print(generate_name("Data Science"))