Prathamesh1420 commited on
Commit
885c166
1 Parent(s): 778eb59

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +23 -23
app.py CHANGED
@@ -1,24 +1,24 @@
1
- import json
2
- from llama_index.core.tools import QueryEngineTool, ToolMetadata
3
- from llama_index.core.agent import ReActAgent
4
- from llama_index.llms.ollama import Ollama
5
- from llama_index.core import VectorStoreIndex, SimpleDirectoryReader, Settings
6
- from llama_index.embeddings.huggingface import HuggingFaceEmbedding
7
- from llama_index.llms.ollama import Ollama
8
- from llama_index.core import Settings
9
- from llama_index.llms.groq import Groq
10
- import os
11
- documents = SimpleDirectoryReader("data").load_data()
12
-
13
- # bge-base embedding model
14
- Settings.embed_model = HuggingFaceEmbedding(model_name="BAAI/bge-base-en-v1.5")
15
-
16
- # ollama
17
- Settings.llm =Groq(model="mixtral-8x7b-32768", api_key="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")
18
-
19
- index = VectorStoreIndex.from_documents(
20
- documents,
21
- )
22
- query_engine = index.as_query_engine()
23
- response = query_engine.query("What did the author do growing up?")
24
  print(response)
 
1
+ import json
2
+ from llama_index.core.tools import QueryEngineTool, ToolMetadata
3
+ from llama_index.core.agent import ReActAgent
4
+ from llama_index.llms.ollama import Ollama
5
+ from llama_index.core import VectorStoreIndex, SimpleDirectoryReader, Settings
6
+ from llama_index.embeddings.huggingface import HuggingFaceEmbedding
7
+ from llama_index.llms.ollama import Ollama
8
+ from llama_index.core import Settings
9
+ from llama_index.llms.groq import Groq
10
+ import os
11
+ documents = SimpleDirectoryReader("data").load_data()
12
+
13
+ # bge-base embedding model
14
+ Settings.embed_model = HuggingFaceEmbedding(model_name="BAAI/bge-base-en-v1.5")
15
+
16
+ # ollama
17
+ Settings.llm =Groq(model="mixtral-8x7b-32768", api_key="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")
18
+
19
+ index = VectorStoreIndex.from_documents(
20
+ documents,
21
+ )
22
+ query_engine = index.as_query_engine()
23
+ response = query_engine.query("What did the author do growing up?")
24
  print(response)