404
Page Not Found
The specified file was not found on this website. Please check the URL for mistakes and try again.
Why am I seeing this?
This page was generated by the Firebase Command-Line Interface. To modify it, edit the 404.html file in your project's configured public directory.
```
### functions/llmPreprocess.js
```javascript
import admin from './firebase.js';
import { generateKeywordPhrasesPrompt, generateQuestionPrompt, generateEventPrompt, generateBrainstormPrompt } from './prompts.js';
import { Client } from '@elastic/elasticsearch';
import { createDoc, searchKeyPhrase } from './rag.js';
import { askMistral } from './mistral.js';
const db = admin.firestore();
async function getJournals(user) {
try {
let journals = await db.collection('journals').where('user', '==', user).get();
let journalsArray = [];
journals.forEach(journal => {
journalsArray.push(journal.data());
});
return journalsArray;
}
catch (error) {
console.log("error with getting journals");
console.log(error);
}
}
// creates a string with all previous journal entries
export async function preprocessJournalsLLM(userId) {
const journals = await getJournals(userId);
// between each journal should be the title of the entry as well
const journalString = journals.map(journal => `${journal.title}\n${journal.text}`).join("\n\n");
return journalString;
}
export async function prepareUserInfo(userId){
try {
const userDoc = await db.collection('users').doc(userId).get();
const user = userDoc.data();
const userInfo = `User Info. name: ${user.name}, date of birth: ${user.dob}, pronouns: ${user.pronouns}, city: ${user.city}, interests: ${user.interests}, bio: ${user.bio}`;
return userInfo;
}
catch (error) {
console.log("error with getting user info");
console.log(error);
}
}
export async function getUserBrainstormPrompt(userId){
const userInfo = await prepareUserInfo(userId);
const journalString = await preprocessJournalsLLM(userId);
let prompt = generateBrainstormPrompt + "\n\n" + userInfo + "\n\n" + "User journal entries:\n" + journalString;
// parse out ```json and ``` from the string
prompt = prompt.replace("```json", "").replace("```", "");
return prompt;
}
```
### functions/users.js
```javascript
import express from 'express';
import cors from 'cors';
import admin from './firebase.js';
import { user } from 'firebase-functions/v1/auth';
import { getUserBrainstormPrompt } from './llmPreprocess.js';
import { askMistral } from './mistral.js';
const users = express()
users.use(cors({origin: true}))
const db = admin.firestore()
users.post('/createUser', async (req, res) => {
try{
await db.collection('users').doc(req.body.uid).set(req.body)
res.status(200).json({message: "user created successfully!"})
} catch(error){
res.status(500).json(error)
}
})
// Read all the users
users.get('/getUsers', (req, res) => {
db.collection('users').get()
.then(snapshot => {
const items = [];
snapshot.forEach(doc => items.push({ id: doc.id, ...doc.data() }));
res.status(200).json(items);
})
.catch(error => res.status(500).json({ error: error.message }));
});
// Read a single user
users.post('/getUser/', (req, res) => {
const itemId = req.body.uid;
db.collection('users').doc(itemId).get()
.then(doc => {
if (!doc.exists) {
res.status(404).json({ error: 'Item not found' });
} else {
res.status(200).json({ id: doc.id, ...doc.data() });
}
})
.catch(error => res.status(500).json({ error: error.message }));
});
// Update a single user
users.post('/updateUser', (req, res) => {
const itemId = req.body.user.uid;
const updatedItem = req.body.user;
console.log("Reached users REST API!")
console.log(updatedItem)
console.log(itemId)
db.collection('users').doc(itemId).update(updatedItem)
.then(() => res.status(200).json({ message: 'Item updated successfully' }))
.catch(error => res.status(500).json({ error: error.message }));
});
// Delete a single user
users.delete('/deleteUser', (req, res) => {
const itemId = req.body.uid;
db.collection('users').doc(itemId).delete()
.then(() => res.status(200).json({ message: 'Item deleted successfully' }))
.catch(error => res.status(500).json({ error: error.message }));
});
users.get("/getUsersCarousel/", async (req, res) => {
try{
let prompt = await getUserBrainstormPrompt(req.query.user);
console.log("prompt: ", prompt);
let response = await askMistral(prompt);
console.log("response 123: ", response);
// parse out ```json and ``` from the string
response = response.replace("```json", "").replace("```", "");
let prompts = JSON.parse(response);
console.log("prompts: ", prompts);
res.status(200).json(prompts);
} catch(error){
res.status(500).json(error);
console.log("error: ", error);
}
});
export { users }
```
### functions/rag.js
```javascript
import { Mistral } from "@mistralai/mistralai";
import { Client } from '@elastic/elasticsearch';
import dotenv from 'dotenv';
import { dirname } from 'path';
import { fileURLToPath } from 'url';
import path from 'path';
const __dirname = dirname(fileURLToPath(import.meta.url));
dotenv.config({ path: path.join(__dirname, '../secrets/.env') });
const mistral = new Mistral({
apiKey: process.env.MISTRAL_API_KEY ?? "",
});
const elasticClient = new Client({
node: process.env.ELASTIC_ENDPOINT ?? "",
auth: {
apiKey: process.env.ELASTIC_API_ENTRIES ?? "",
}
});
async function getEmbeddings(text){
try{
const result = await mistral.embeddings.create({
inputs: [text],
model: "mistral-embed",
});
const mistralEmbedding = result.data[0].embedding;
// Handle the result
return mistralEmbedding;
} catch (error) {
console.log("error with getting embedding");
console.log(error);
}
}
export async function createDoc(firebaseId, text, userId){
const embeddingResponse = await getEmbeddings(text);
console.log("gotten embedding: ", embeddingResponse);
const documentWithEmbedding = {
text: text,
embedding: embeddingResponse,
id: firebaseId,
userId: userId,
};
console.log("document with embedding:");
console.log(documentWithEmbedding);
try {
const res = await elasticClient.index({
index: "j2",
document: documentWithEmbedding,
});
}
catch (error) {
console.log("error with creating new vector in elastic");
console.log(error);
}
console.log("entry created successfully!");
}
export async function searchKeyPhrase(keyphrase, threshold, userId) {
const keyphraseVector = await getEmbeddings(keyphrase);
try {
const searchResponse = await elasticClient.search({
index: "j2",
body: {
query: {
script_score: {
query: {
// bool: {
// must: [
// { term: { userId: userId } },
// { exists: { field: "embedding" } }
// ]
// }
exists: { field: "embedding" }
},
script: {
source: "cosineSimilarity(params.query_vector, 'embedding')",
params: { query_vector: keyphraseVector }
}
}
},
size: 100 // Limit to top 5 most similar results
}
});
const hits = searchResponse.hits.hits;
console.log("Number of hits:", hits.length);
if (hits.length == 0) {
return;
}
const journals = hits.map(hit => ({
...hit._source,
score: hit._score
}));
// find if sim score is > threshold
if (journals && journals[0].score > threshold){
console.log("Found journal with score above threshold");
return journals[0].id;
}
else{
console.log("sim score below threshold", journals[0].score);
return;
}
} catch(error) {
console.error('Search error:', error);
}
}
```
[3 more indexed source files omitted to keep this export small. The full file list is in the Codebase structure section above.]