Example of Gemini - Jaxx - VectorDB.ipynb

{"cells":[{"cell_type":"code","execution_count":null,"metadata":{"colab":{"background_save":true},"id":"ALpV_ZyVae1h"},"outputs":[],"source":["pip install -q google-generativeai"]},{"cell_type":"code","execution_count":null,"metadata":{"id":"EUDDDTnfae1i"},"outputs":[],"source":["from main import VectorDB\n","import google.generativeai as genai"]},{"cell_type":"code","execution_count":null,"metadata":{"id":"Z0DcvROBae1j"},"outputs":[],"source":["genai.configure(api_key=\"YOUR_API_KEY\")\n","\n","# Set up the model\n","generation_config = {\n","  \"temperature\": 1,\n","  \"top_p\": 0.95,\n","  \"top_k\": 0,\n","  \"max_output_tokens\": 8192,\n","}\n","\n","safety_settings = [\n","  {\n","    \"category\": \"HARM_CATEGORY_HARASSMENT\",\n","    \"threshold\": \"BLOCK_NONE\"\n","  },\n","  {\n","    \"category\": \"HARM_CATEGORY_HATE_SPEECH\",\n","    \"threshold\": \"BLOCK_NONE\"\n","  },\n","  {\n","    \"category\": \"HARM_CATEGORY_SEXUALLY_EXPLICIT\",\n","    \"threshold\": \"BLOCK_NONE\"\n","  },\n","  {\n","    \"category\": \"HARM_CATEGORY_DANGEROUS_CONTENT\",\n","    \"threshold\": \"BLOCK_NONE\"\n","  },\n","]"]},{"cell_type":"code","execution_count":null,"metadata":{"id":"qtwg8z6gae1j"},"outputs":[],"source":["model = genai.GenerativeModel(model_name=\"gemini-1.5-pro-latest\",\n","                              generation_config=generation_config,\n","                              safety_settings=safety_settings)\n","\n","convo = model.start_chat(history=[\n","])"]},{"cell_type":"code","execution_count":null,"metadata":{"id":"Auv8y-2kae1j"},"outputs":[],"source":["vector_db = VectorDB(model_name='BAAI/bge-base-en-v1.5')"]},{"cell_type":"code","execution_count":null,"metadata":{"id":"zMimI8fqae1k"},"outputs":[],"source":["texts = [\"\"\"Pizza (/ˈpiːtsə/ PEET-sə, Italian: [ˈpittsa]; Neapolitan: [ˈpittsə]) is a traditional Italian dish typically consisting of a flat base of leavened wheat-based dough topped with tomato, cheese, and other ingredients, baked at a high temperature, traditionally in a wood-fired oven.[1] The term pizza was first recorded in the year 997 AD, in a Latin manuscript from the southern Italian town of Gaeta, in Lazio, on the border with Campania.[2] Raffaele Esposito is often credited for creating modern pizza in Naples.[3][4][5][6] In 2009, Neapolitan pizza was registered with the European Union as a traditional speciality guaranteed dish. In 2017, the art of making Neapolitan pizza was added to UNESCO's list of intangible cultural heritage.[7] Pizza and its variants are among the most popular foods in the world. Pizza is sold at a variety of restaurants, including pizzerias (pizza specialty restaurants), Mediterranean restaurants, via delivery, and as street food.[8] In Italy, pizza served in a restaurant is presented unsliced, and is eaten with the use of a knife and fork.[9][10] In casual settings, however, it is typically cut into slices to be eaten while held in the hand. Pizza is also sold in grocery stores in a variety of forms, including frozen or as kits for self-assembly. They are then cooked using a home oven. In 2017, the world pizza market was US$128 billion, and in the US it was $44 billion spread over 76,000 pizzerias.[11] Overall, 13% of the U.S. population aged two years and over consumed pizza on any given day.[12]\"\"\",\n","\"\"\"Chess is a board game for two players. It is sometimes called international chess or Western chess to distinguish it from related games such as xiangqi (Chinese chess) and shogi (Japanese chess). Chess is an abstract strategy game that involves no hidden information and no elements of chance. It is played on a chessboard with 64 squares arranged in an 8×8 grid. The players, referred to as \"White\" and \"Black\", each control sixteen pieces: one king, one queen, two rooks, two bishops, two knights, and eight pawns. White moves first, followed by Black. The game is won by checkmating the opponent's king, i.e. threatening it with inescapable capture. There are several ways a game can end in a draw. The recorded history of chess goes back at least to the emergence of a similar game, chaturanga, in seventh-century India. The rules of chess as they are known today emerged in Europe at the end of the 15th century, with standardization and universal acceptance by the end of the 19th century. Today, chess is one of the world's most popular games, and is played by millions of people worldwide. Organized chess arose in the 19th century. Chess competition today is governed internationally by FIDE (Fédération Internationale des Échecs; the International Chess Federation). The first universally recognized World Chess Champion, Wilhelm Steinitz, claimed his title in 1886; Ding Liren is the current World Champion. A huge body of chess theory has developed since the game's inception. Aspects of art are found in chess composition, and chess in its turn influenced Western culture and the arts, and has connections with other fields such as mathematics, computer science, and psychology. One of the goals of early computer scientists was to create a chess-playing machine. In 1997, Deep Blue became the first computer to beat the reigning World Champion in a match when it defeated Garry Kasparov. Today's chess engines are significantly stronger than the best human players and have deeply influenced the development of chess theory; however, chess is not a solved game.\"\"\",\n","\"\"\"A bus (contracted from omnibus,[1] with variants multibus, motorbus, autobus, etc.) is a road vehicle that carries significantly more passengers than an average car or van, but less than the average rail transport. It is most commonly used in public transport, but is also in use for charter purposes, or through private ownership. Although the average bus carries between 30 and 100 passengers, some buses have a capacity of up to 300 passengers.[2] The most common type is the single-deck rigid bus, with double-decker and articulated buses carrying larger loads, and midibuses and minibuses carrying smaller loads. Coaches are used for longer-distance services. Many types of buses, such as city transit buses and inter-city coaches, charge a fare. Other types, such as elementary or secondary school buses or shuttle buses within a post-secondary education campus, are free. In many jurisdictions, bus drivers require a special large vehicle licence above and beyond a regular driving licence. Buses may be used for scheduled bus transport, scheduled coach transport, school transport, private hire, or tourism; promotional buses may be used for political campaigns and others are privately operated for a wide range of purposes, including rock and pop band tour vehicles. Horse-drawn buses were used from the 1820s, followed by steam buses in the 1830s, and electric trolleybuses in 1882. The first internal combustion engine buses, or motor buses, were used in 1895.[3] Recently, interest has been growing in hybrid electric buses, fuel cell buses, and electric buses, as well as buses powered by compressed natural gas or biodiesel. As of the 2010s, bus manufacturing is increasingly globalised, with the same designs appearing around the world.\"\"\"]\n","vector_db.add_texts(texts)"]},{"cell_type":"code","execution_count":null,"metadata":{"id":"I_oKG4GAae1k"},"outputs":[],"source":["text_query = \"I like playing Chess, tell me more about it.\"\n","text_results = vector_db.search(text_query, top_k=1, type='text')\n","print(\"Text search results:\")\n","for text, similarity in text_results:\n","    result = text\n","    print(f\"Text: {text}, Similarity: {similarity}\")"]},{"cell_type":"code","execution_count":null,"metadata":{"id":"ln1UrvzGae1l"},"outputs":[],"source":["convo.send_message(\"Question: \" + text_query + \" Context: \" + result)\n","print(convo.last.text)"]}],"metadata":{"colab":{"provenance":[]},"kernelspec":{"display_name":"Python 3","name":"python3"},"language_info":{"name":"python"}},"nbformat":4,"nbformat_minor":0}