Skip to content

Commit 79934ae

Browse files
exowandererrti
authored andcommitted
renewed app.mount; bug fixed PEP8 changes in api.py; reformatted rag.py inputs; tests timed out
1 parent 86abc9b commit 79934ae

File tree

4 files changed

+26
-12
lines changed

4 files changed

+26
-12
lines changed

frontend/src/views/ChatView.vue

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -97,7 +97,7 @@ const inputFocused = ref(false)
9797
function search() {
9898
response.value = undefined;
9999
displayResponse.value = true
100-
fetch(`/api?q=${inputText.value}`)
100+
fetch(`/api?query=${inputText.value}`)
101101
.then((response) => response.json())
102102
.then((data) => {
103103
response.value = data

gswikichat/api.py

Lines changed: 22 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -2,10 +2,28 @@
22
from fastapi.staticfiles import StaticFiles
33
from fastapi import FastAPI
44

5-
from .rag import embedder, retriever, prompt_builder, llm, answer_builder
5+
from .rag import answer_builder
6+
from .llm_config import llm
7+
from .prompt import prompt_builders
8+
from .vector_store_interface import embedder, retriever, input_documents
9+
610
from haystack import Document
711

8-
homepage = "/frontend/dist"
12+
import logging
13+
import sys
14+
15+
logger = logging.getLogger()
16+
logger.setLevel(logging.DEBUG)
17+
18+
handler = logging.StreamHandler(sys.stdout)
19+
handler.setLevel(logging.DEBUG)
20+
formatter = logging.Formatter(
21+
'%(asctime)s - %(name)s - %(levelname)s - %(message)s')
22+
handler.setFormatter(formatter)
23+
logger.addHandler(handler)
24+
25+
static_dir = 'frontend/dist'
26+
homepage = f'/{static_dir}'
927

1028
app = FastAPI()
1129
app.mount(
@@ -38,7 +56,7 @@ async def api(query, top_k=3, lang='en'):
3856
query_embedded = embedder.run([query])
3957
query_embedding = query_embedded['documents'][0].embedding
4058

41-
retreiver_results = retriever.run(
59+
retriever_results = retriever.run(
4260
query_embedding=list(query_embedding),
4361
filters=None,
4462
top_k=top_k,
@@ -47,7 +65,7 @@ async def api(query, top_k=3, lang='en'):
4765
)
4866

4967
logger.debug('retriever results:')
50-
for retriever_result in retriever_results:
68+
for retriever_result_ in retriever_results:
5169
logger.debug(retriever_result_)
5270

5371
prompt_builder = prompt_builders[lang]

gswikichat/prompt.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -49,7 +49,7 @@
4949
# {% endfor %}
5050
# """
5151

52-
prompt_builder = {
53-
'en': PromptBuilder(template=prompt_template_en),
54-
'de': PromptBuilder(template=prompt_template_de),
52+
prompt_builders = {
53+
'en': PromptBuilder(template=prompt_template_en),
54+
'de': PromptBuilder(template=prompt_template_de),
5555
}

gswikichat/rag.py

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -2,10 +2,6 @@
22
from haystack import Pipeline
33
from haystack.components.builders.answer_builder import AnswerBuilder
44

5-
from .llm_config import llm
6-
from .prompt import prompt_builder
7-
from .vector_store_interface import embedder, retriever, input_documents
8-
95
answer_builder = AnswerBuilder()
106

117
# rag_pipeline = Pipeline()

0 commit comments

Comments
 (0)