bupa1018 commited on
Commit
0486180
·
1 Parent(s): 4b58dd9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -29
app.py CHANGED
@@ -88,35 +88,6 @@ def setup_llm(model_name, temperature, api_key):
88
  llm = ChatGroq(model=model_name, temperature=temperature, api_key=api_key)
89
  return llm
90
 
91
-
92
- def format_kadi_apy_library_context(docs):
93
- doc_context = []
94
-
95
- for doc in docs:
96
- # Extract metadata information
97
- class_info = doc.metadata.get("class", "Unknown Class")
98
- type_info = doc.metadata.get("type", "Unknown Type")
99
- source_info = doc.metadata.get("source", "Unknown Type")
100
-
101
- print(":}\n\n", doc.page_content)
102
- formatted_doc = f"# source: {source_info}\n# class: {class_info}\n# type: {type_info}\n{doc.page_content}\n\n\n"
103
- doc_context.append(formatted_doc)
104
-
105
- return doc_context
106
-
107
-
108
- def format_kadi_api_doc_context(docs):
109
- doc_context = []
110
-
111
- for doc in docs:
112
- source_info = doc.metadata.get("source", "Unknown Type")
113
- print(":}\n\n", doc.page_content)
114
- formatted_doc = f"# source: {source_info}\n{doc.page_content}\n\n\n"
115
- doc_context.append(formatted_doc)
116
-
117
- return doc_context
118
-
119
-
120
 
121
  def rag_workflow(query):
122
  """
 
88
  llm = ChatGroq(model=model_name, temperature=temperature, api_key=api_key)
89
  return llm
90
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
91
 
92
  def rag_workflow(query):
93
  """